Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix recording bug in after_train function #600

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

Coolshanlan
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant