Skip to content
Snippets Groups Projects
Unverified Commit 99de0951 authored by Mashiro's avatar Mashiro Committed by GitHub
Browse files

[Enhance] EMAHook support does not load checkpoint strictly (#352)

* BaseAveragedModel support load ckpt without module prefix

* refine docstring

* allow EMAHook does not load ckpt strictly

* add unit test for strict argument of EMAHook

* sync remote

* sync remote

* clean the code

* ema hook supports setting start iter

* fix unit test

* fix as comment

* fix as comment

* describe kwargs
parent 08602a23
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment