search

WarBean / tps_stn_pytorch

PyTorch implementation of Spatial Transformer Network (STN) with Thin Plate Spline (TPS)
926 stars 155 forks source link

Update mnist_visualize.py #16

Open freepoet opened 3 years ago

freepoet commented 3 years ago

index shouldn't be a tensor,it should be int type, otherwise, it will report a error:

KeyError: tensor(x)