Skip to content

wanglei666666666/pytorch-siamese

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-siamese

This is a port of chainer-siamese

Install

This installation requires cuda to be installed.

$ virtualenv /usr/bin/python3.5 env
$ env/bin/pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
$ env/bin/pip install torchvision

Run

$ env/bin/python train_mnist.py --epoch 10

This dumps for every epoch ther current state and creates a result.png.

Run specific model

$ env/bin/python train_mnist.py -m model-epoch-7.pth

Result

About

Siamese Network implementation using Pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%