run the training on wider input images (width=height>>28)

[kao123]

great project!
i succeeded to run the code on my own images with input size=28.
I was trying to run the code using a different input size (e.g widht=height=300)
As soon as i modify the args.image_height = args.image_width to any other value than 28 (in my data_loader and mnist_train)
i get the following error

File "/home/myaccount/tps_stn_pytorch/tps_grid_gen.py", line 67, in forward
assert source_control_points.size(1) == self.num_points
AssertionError
'''

I tried to modify the tps_grid_gen code.. but nothing's worked.
Any help please

[KakaVlasic]

hello! I have the same problem, how do you deal with it? I try to realign the code in mnist_train 'def train(epoch)' , the AssertionError disappear but i don't get any output(checkpoint, accuracy log). but the code indeed run on my gpu.
any help would be appreciated!

[ilyalasy]

Hey, is this problem solved?
@WarBean

[Fleyderer]

This problem is not solved yet :\