Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
Repository files navigation
1. Prerequisites python 3.7 pytorch 1.1 tqdm sklearn tensorboardX CUDA Toolkit 11+ 2. Open MAIN.py -Setting parameters such as running modes, save path, and other hyperparameters, to define configuration. -Setting TRAIN=1 to train the model. -Setting EVAL=1 to test the model in terms of effectiveness and efficiency.