Add move_to_device kwarg to the optimizer's load_state_dict
#665
| Job | Run time |
|---|---|
| 3m 2s | |
| 3m 2s |
move_to_device kwarg to the optimizer's load_state_dict
#665
| Job | Run time |
|---|---|
| 3m 2s | |
| 3m 2s |