DBNet.pytorch icon indicating copy to clipboard operation
DBNet.pytorch copied to clipboard

关于Best model是怎么评判的

Open wwp-max opened this issue 3 years ago • 2 comments

image 请问这里的bestmodel是根据什么选择的呢,我看并不是完全和精确度一样。

wwp-max avatar Apr 15 '22 16:04 wwp-max

在 trainer 中 trainer.py 的第172~176行 根據 metrics 的 train_loss 大小來儲存

t107598066 avatar Apr 22 '22 06:04 t107598066

  • 根据hmean的值来保存最优模型的 https://github.com/WenmuZhou/DBNet.pytorch/blob/678b2ae55e018c6c16d5ac182558517a154a91ed/trainer/trainer.py#L181-L186

SWHL avatar Jun 08 '22 10:06 SWHL