• Python 3.8
• Pytorch 1.8.1
• CUDA 12.1

1. THUMOS'14

   • use the features and annotations provided by this repo.

2. Place the features and annotations inside a dataset/Thumos14reduced/ folder.

You can easily train the model by running the provided script.

• Refer to options.py. Modify the argument of dataset-root to the path of your dataset folder.

• Run the command below.

$ python main.py --run-type 0 --model-id 1

Models are saved in ./ckpt/dataset_name/model_id/

The trained model can be found here. (This saved model's result is slightly different from the one reported in our paper.)

Please put it into ./ckpt/dataset_name/model_id/.

• Run the command below.

$ python main.py --pretrained --run-type 1 --model-id 1

Please refer to the log in the same folder of saved models to set the load epoch of the best model. Make sure you set the right model-id that corresponds to the model-id during training.