search

ZikangZhou / QCNet

[CVPR 2023] Query-Centric Trajectory Prediction
https://openaccess.thecvf.com/content/CVPR2023/papers/Zhou_Query-Centric_Trajectory_Prediction_CVPR_2023_paper.pdf
Apache License 2.0
432 stars 70 forks source link

How to load checkpoint and continue training? #39

Open CoconutSweet999 opened 3 months ago

CoconutSweet999 commented 3 months ago

I wonder how to load checkpoint and continue training. Because of the OOM (maybe at 30 epochs), I have to load checkpoint and continue training after OOM. I try to add ckpt_path = '....ckpt' (path from lightning_logs/checkpoint), but it seems useless. How should I solve this problem？ Thanks a lot!

SunHaoOne commented 3 months ago

For example, you can have a try to use different lr or use the default lr.

model = QCNet.load_from_checkpoint(checkpoint_path=ckpt_path)
model = QCNet.load_from_checkpoint(checkpoint_path=ckpt_path, lr=lr)