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

[Fix] Fix missing device ids in wrap_model (#351) 

* fix missing device ids in wrap_model

* clean the code

* use default broadcast_buffers

* refine MMSeparateDistributedDataParallel

* rename tmp variable

* refine docstring

* add type hints

* refactor docstring of ddp model

* add  arg in docstring

* minor refine

* better ddp link
parent 792f481e
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