Skip to content
Snippets Groups Projects
Unverified Commit 2086bc45 authored by Jiazhen Wang's avatar Jiazhen Wang Committed by GitHub
Browse files

[Feature] Fully support to use MLU for training (#313)

* modify cuda() to to()

* rollback load_checkpoint

* refine runner

* add TODO
parent afeac1c0
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