search

sming256 / OpenTAD

OpenTAD is an open-source temporal action detection (TAD) toolbox based on PyTorch.
Apache License 2.0
106 stars 5 forks source link

AdaTAD Training Resume #11

Closed tqosu closed 1 month ago

tqosu commented 1 month ago

Hi Shuming,

Will the training automatically resume if I kill the process and run it again?

Thanks.

sming256 commented 1 month ago

We support resuming the training from a checkpoint.

For example, to resume the training of AdaTAD on THUMOS, you can use the following command torchrun --nnodes=1 --nproc_per_node=2 --rdzv_backend=c10d --rdzv_endpoint=localhost:0 tools/train.py configs/adatad/thumos/e2e_thumos_videomae_s_768x1_160_adapter.py --resume exps/thumos/adatad/e2e_actionformer_videomae_s_768x1_160_adapter/gpu2_id0/epoch_21.pth

tqosu commented 1 month ago

Thanks.