S0: Create environment (Recommend Python 3.6)
S1: Install pytorch with tutorial from pytorch.org
.
S2: Install requirements.txt
by pip install -r requirements.txt
First, go to source
folder.
Call:
python main.py <path to train folder> --test_folder <path to test folder>
Check tensorboard by:
tensorboard --logdir=lightning_logs