zhangwei

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