Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/delijati/pytorch-siamese
Siamese Network implementation using Pytorch
https://github.com/delijati/pytorch-siamese
contrastive pytorch siamese
Last synced: 5 days ago
JSON representation
Siamese Network implementation using Pytorch
- Host: GitHub
- URL: https://github.com/delijati/pytorch-siamese
- Owner: delijati
- License: mit
- Created: 2017-05-05T15:57:05.000Z (almost 8 years ago)
- Default Branch: master
- Last Pushed: 2023-06-20T13:33:11.000Z (over 1 year ago)
- Last Synced: 2025-01-30T03:11:39.761Z (13 days ago)
- Topics: contrastive, pytorch, siamese
- Language: Python
- Size: 177 KB
- Stars: 277
- Watchers: 5
- Forks: 65
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# pytorch-siamese
This is a port of [chainer-siamese](https://github.com/mitmul/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
