def save_model(self, save_path):
data = L.get_all_param_values(self.network)
with open(save_path, 'w') as f:
pkl.dump(data, f)
微信
公众号
扫码关注公众号