PyTorch_Tutorial icon indicating copy to clipboard operation
PyTorch_Tutorial copied to clipboard

Published 20 hours ago • verified publisher icon TingsongYu

建议

Open goaheand opened this issue 5 years ago • 0 comments

在_viewer/_visual_weights 中，不需要重写Net再加载net_params.pkl的state_dict啊，直接 pre_trained_dict = torch.load('../2_model/net_params.pkl') 就已经得到想要的state_dict了。 建议删除冗余的部分

goaheand avatar Mar 01 '19 02:03 goaheand