Skip to content

Latest commit

 

History

History
10 lines (7 loc) · 618 Bytes

README.md

File metadata and controls

10 lines (7 loc) · 618 Bytes
Raw

CODE FOR MOVES

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}