mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

init OptimWrapper after resume (#362)

Browse Source
This commit is contained in:
Mashiro 2022-07-14 20:10:48 +08:00 committed by GitHub
parent 45001a1f6f
commit c9c6d454f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmengine/runner/runner.py
View File

@ -1592,16 +1592,17 @@ class Runner:
self._val_loop) # type: ignore self._val_loop) # type: ignore
self.call_hook('before_run') self.call_hook('before_run')
# TODO: add a contextmanager to avoid calling `before_run` many times
# make sure checkpoint-related hooks are triggered after `before_run`
self.load_or_resume()
# Initiate inner count of `optim_wrapper`. # Initiate inner count of `optim_wrapper`.
self.optim_wrapper.initialize_count_status( self.optim_wrapper.initialize_count_status(
self.model, self.model,
self._train_loop.iter, # type: ignore self._train_loop.iter, # type: ignore
self._train_loop.max_iters) # type: ignore self._train_loop.max_iters) # type: ignore
# TODO: add a contextmanager to avoid calling `before_run` many times
# make sure checkpoint-related hooks are triggered after `before_run`
self.load_or_resume()
self.train_loop.run() # type: ignore self.train_loop.run() # type: ignore
self.call_hook('after_run') self.call_hook('after_run')