建议

[zhangwei]

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