[Feature] Fully support to use MLU for training (#313)
* modify cuda() to to() * rollback load_checkpoint * refine runner * add TODO
Loading
Please register or sign in to comment
* modify cuda() to to() * rollback load_checkpoint * refine runner * add TODO