search

fangpin / siamese-pytorch

Implementation of Siamese Networks for image one-shot learning by PyTorch, train and test model on dataset Omniglot
279 stars 59 forks source link

can I use torch1.6? #13

Closed tkamkb closed 2 years ago

tkamkb commented 2 years ago

the requirement says the torch version is 1.0.1, can I use torch1.6 to train and test the network?

fangpin commented 2 years ago

didn't test it, probably some code change is required to migrate to new torch API.

tkamkb commented 2 years ago

I tested, it works.