pytorch(1.12.1), sklearn, numpy, python 3.8+
- Save the pre-trained model under the checkpoint folder, Download here
- Place the Dataset and adjust the script path variable.
- Go to the scripts folder
source train_rvt.sh
The script will automatically implement the Leave One Out Cross-Validation(LOOCV).
The running log file train.log will be output in log folder.
The experiment results are output in output_rvt.json in output folder