Castor

Pytorch deep learning models.

1. SM model: Similarity between question and candidate answers.

Setting up Pytorch

You need Python 3.6 to use the models in this repository.

As per pytorch.org

Anaconda is our recommended package manager

conda install pytorch torchvision -c soumith

Other pytorch installation modalities (e.g. via pip) can be seen at pytorch.org.

We also recommend gensim. We use some gensim modules to cache word embeddings.

conda install gensim

Pytorch has good support for GPU computations. CUDA installation guide for linux can be found here

NOTE: Install CUDA libraries before installing conda and pytorch.

Castor-data

Sourcing and pre-processing of input data for each model is described in respective model/README.md's