Jiayao Tan, Fan Lyu, Linyan Li, Fuyuan Hu, Tingliang Feng, Fenglei Xu, Zhang Zhang, Rui Yao, Liang Wang
We follow DiscoNet setting to use the same data index_list for training. Please follow the guidelines in DiscoNet to prepare them.
python train_codet.py
--data /path/to/training/dataset
--com disco
--log --batch 4
--kd_flag 1
--resume_teacher /path/to/teacher/checkpoint.pth
--auto_resume_path logs
--logpath logs
--nepoch 100
-- rsu [0/1]
python test_codet.py
--data /path/to/testing/dataset
--com disco
--resume /path/to/teacher/checkpoint.pth
--tracking
--logpath logs
--visualization 1
--rsu 1
Our project references the codes in the following repos.