Я хочу использовать STN в своих данных для исправления тета-параметра, я надеюсь, что после обучения все изображения от lo cnet до STN могут быть направлены в одном направлении (как номер MNIST в статье вверх) , как мне спроектировать аффинный (тэта-параметр) и lo cnet, прикрепить мой код Cloud здесь, не могли бы вы дать мне несколько советов?
https://drive.google.com/open?id=1r4rEGO7j2nPr46mE0sKJHdaBWkmWKm16
Мой тренировочный процесс часто показывает черный экран, например https://drive.google.com/open?id=1KVk3KfWXV8U7h6iUO0y47kT8CA5tbV8a -> https://drive.google.com/open?id=1S0KskrTllCG8-tMguH4JchbWk9sTfiuF, я понятия не имею, почему?