AdelaiDet icon indicating copy to clipboard operation
AdelaiDet copied to clipboard

fix recording bug in after_train function

Open Coolshanlan opened this issue 2 years ago • 0 comments

When I evaluate model in the after_train function and record results, the result will not be recorded in Tensorboard and metrics.json

This is because detectron2.utils.events.TensorboardXWriter and JSONWriter will check the trainer.iter, and each iter will only record once. If you do not set trainer.iter+1 before executing after_train, the execution result will not be recorded, because the current trainer.iter has been used in the last step

So I fixed this issue in train_loop function

Coolshanlan avatar Dec 06 '22 12:12 Coolshanlan