The training codes for the differnet datasets is available in the separate folder. Simply download the dataset and follow the instruction to start training the model.
For installation
- The code was tested on Ubunut 20.04
- For environment the code was tested with conda to install torch=1.9.0 with cuda 11.1. it may work with some earler version too. Navigate https://pytorch.org/get-started/previous-versions/ and use the appropriate version to install pytorch. Tensorboard may be required to be installed for logging purposes - logs are stored inside runs/test{log_name}