SRCNN
best_weights = copy.deepcopy(model.state_dict())
copy.deepcopy是深拷贝的意思
model.state_dict()
其实返回的是一个OrderDict
,存储了网络结构的名字和对应的参数。
SRCNN
best_weights = copy.deepcopy(model.state_dict())
copy.deepcopy是深拷贝的意思
model.state_dict()
其实返回的是一个OrderDict
,存储了网络结构的名字和对应的参数。