Get Started

pip install -r requirements.txt

python app.py

Train

pip install -r requirements-train.txt

python train_torch.py

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
model		model
runs/experiment		runs/experiment
.gitignore		.gitignore
README.md		README.md
app.py		app.py
config.py		config.py
gpt_handler.py		gpt_handler.py
image_preprocessing.py		image_preprocessing.py
requirements-train.txt		requirements-train.txt
requirements.txt		requirements.txt
torch_datamodule.py		torch_datamodule.py
torch_models.py		torch_models.py
train_torch.py		train_torch.py