Skip to content

Files

Latest commit

 

History

History

nabla_sim_mnist

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

Nabla Sim MNIST

Setup environment

Dependencies can be found in requirements.txt:

  • matplotlib
  • numpy
  • PIL
  • torch
  • torchvision

However the docker/pytorch-netsimilarity Docker image has everything installed so it can be easier to use it. Launch that Docker image with sh run-docker.sh.

Executepython main.py to launch the training.

Then open and run notebooks/mnist_viz.ipynb to visualize the results. You can start a Jupyter notebook from the pytorch-netsimilarity Docker image with sh /start_jupyter.sh.