/cs231n-molecular-translation

Primary LanguagePython

Baseline code copied from https://www.kaggle.com/pasewark/pytorch-resnet-lstm-with-attention.

How to Use

  1. Download the dataset into the input folder.
input/
    - train/
        - 0/
        - ...
    - test/
        - 0/
        - ...
  1. Set up the environment.
conda env create -f environment.yml
conda activate venv

  1. Change the training configs in config.yaml.

  2. Start training.

python train.py