Implementation of "Generating Sentences from a Continuous Space" paper using pytorch. Generates SARS-COV2 Spike proteins based on the GISAID dataset.
- Python 3.9
- Pytorch 0.4.0
GISAID data not included; must be stored locally.
To train the model from the beginning, run:
python main.py
To resume training from a saved model, run:
python main.py --resume_training=True
To generate samples from a saved model, run:
python main.py --to_train=False