Skip to content
This repository was archived by the owner on Jul 1, 2024. It is now read-only.

Commit aad649e

Browse files
isunjinfacebook-github-bot
authored andcommitted
Adding more event log
Summary: Add more event log Reviewed By: mannatsingh Differential Revision: D19669444 fbshipit-source-id: 9eed43710edf7b524719b1d4933c1e4724b63c60
1 parent 183f518 commit aad649e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

classy_vision/trainer/elastic_trainer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,11 @@ def _run_step(self, state, local_variables, use_gpu):
9595

9696
if state.advance_to_next_phase:
9797
logging.info("Begin advance_phase")
98+
self.elastic_coordinator._log_event("before advance_phase")
9899
state.task.advance_phase()
99100
self.elastic_coordinator.barrier()
100101
logging.info("Done advance_phase")
102+
self.elastic_coordinator._log_event("after advance_phase")
101103

102104
# Start phase hooks
103105
state.task.run_hooks(

0 commit comments

Comments
 (0)