how to use checkpoints and log to continue training?

a1600012888 / MUTR3D

Code for Paper, MUTR3D: A Multi-camera Tracking Framework via 3D-to-2D Queries. https://tsinghua-mars-lab.github.io/mutr3d/

MIT License

179 stars 19 forks source link

how to use checkpoints and log to continue training? #9

Closed XuanjieXiao closed 2 years ago

XuanjieXiao commented 2 years ago

what's more, I'm curious how to continue training if my program was stopped by accidents! so how to use checkpoints to continue training? thks! looking forward to your reply !

a1600012888 commented 2 years ago

Hi, Xuanjie

For resume the training, yes you can do that. See here: https://github.com/a1600012888/MUTR3D/blob/main/tools/train_tracker.py#L27

If you just want to load the one of the checkpoint and start a new training schdule, you can change the load_from in config file https://github.com/a1600012888/MUTR3D/blob/main/plugin/track/configs/resnet50_fpn_5frame.py#L270

XuanjieXiao commented 2 years ago

Hi, Xuanjie

For resume the training, yes you can do that. See here: https://github.com/a1600012888/MUTR3D/blob/main/tools/train_tracker.py#L27

If you just want to load the one of the checkpoint and start a new training schdule, you can change the load_from in config file https://github.com/a1600012888/MUTR3D/blob/main/plugin/track/configs/resnet50_fpn_5frame.py#L270

thks！！！