TimeT_clean
October 17, 2024 ยท View on GitHub
This is a better implementation of TimeT model.
Running the Code
-
Main training script:
python exp_time_tuning_v2.pyYou can customize the run with command-line arguments:
python exp_time_tuning_v2.py --device cuda:0 --batch_size 32 --num_epochs 100 -
For evaluation:
python evaluation.py -
To test the feature forwarder:
python test_feature_forwarder.py
Dataset Files
The train.txt and val.txt files contain lists of video identifiers for the training and validation sets respectively. These files are used to split the dataset into training and validation subsets.
Logging and Visualization
The code uses Weights & Biases (wandb) for logging and visualization. Ensure you're logged in to your wandb account.
Performance curves
You should see something like this:

Where the evaluation is done on PascalVOC.