资源分类
》 资源 》算法》pytorch-siamese

pytorch-siamese

2020-01-13 | |  41 |   0 |   0

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


上一篇：SiameseSentenceSimilarity

下一篇：Siamese-LSTM

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...