tps_stn_pytorch
tps_stn_pytorch copied to clipboard
Update mnist_visualize.py
index shouldn't be a tensor,it should be int type, otherwise, it will report a error:
KeyError: tensor(x)