Add move_to_device
kwarg to the optimizer's load_state_dict
(#1344)
#159
Loading
move_to_device
kwarg to the optimizer's load_state_dict
(#1344)
#159