istn
istn copied to clipboard
Error when trained
I run :
python istn-reg.py --config data/synth2d/config.json --transformation affine --loss u --out output/stn-u --model output/stn-u/train/model
I get the error :
empty(): argument 'size' must be tuple of ints, but found element of type Tensor at pos 2
hello, I met the same problem, all of packages have been installed,but the net still do not work. did you solve it?