Add move_to_device kwarg to the optimizer's load_state_dict (#1344)
#159
| Job | Run time |
|---|---|
| 2m 48s | |
| 2m 48s |
move_to_device kwarg to the optimizer's load_state_dict (#1344)
#159
| Job | Run time |
|---|---|
| 2m 48s | |
| 2m 48s |