How to test the model?
ZhengLeon opened this issue · 8 comments
ZhengLeon commented
@DoubtedSteam
Hi~
How can I eval the model and get the result in your paper after training?
Thanks!
DoubtedSteam commented
set epoch to 0 and resume the checkpoint
you can add the resume: path/to/the/checkpoint into yaml file
ZhengLeon commented
Thanks a lot!
If I enable the resume in training process, how can I update the begining epoch num using the resume one instead of epoch0?
DoubtedSteam commented
edit num_epoch in yaml file
ZhengLeon commented
DoubtedSteam commented
ZhengLeon commented
In Chinese:
恢复训练时的epoch是从0开始的,这样似乎会覆盖之前的checkpoint,如何修改使重新训练的epoch是从resume的轮数开始呢?
DoubtedSteam commented
I didn't achieve this function.
You can add a start_epoch parameter for def get_trainer() in engine/init.py
实际上太麻烦了,因为学习率的step要一次次调用,所以没写这个功能
ZhengLeon commented
strategy_cfg.num_epoch = 60
num_epoch: 140
Dose only the second one make sense?
嗯嗯 如果训练过程中断了epoch的数量就只能从0开始比较方便吗