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

[Fix] Regist pytorch ddp and dp to `MODEL_WRAPPERS`, add unit test to `is_model_wrapper` (#474) 

* regist pytorch ddp and dp, add unit test

* minor refine

* Support check custom wrapper

* enhance ut
parent d0a74f9a
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Datenschutz und Impressum