fangpin / siamese-pytorch

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

add requirements.txt, fix some typos and a small bug in the training script #7

Closed urbanophile closed 5 years ago

urbanophile commented 5 years ago

Thank you for making this code available. I tried to add expand a little in the readme, fix a couple a couple of typos, and there was a small bug concerning access zero dimensional tensors which I also changed. Hopefully the code will work a little easier out of the box with these changes.

fangpin commented 5 years ago

Thanks for your contribution.