Adding assign to load_state_dict implementations
#3080
+90
−9
Loading