Hello Ptrblck,
Thanks for the information, i could not able to find the pretrained weights  , the file named train.pt not available, could you please update further on this if you have done this project.
python train.pt --test-init True --test-epoch 10 --output-dir <path_to_folder_with_snapshots>
Thanks,
Suhas