def save_model(self, save_path):
with open(save_path, 'w') as f:
data = L.get_all_param_values(self.network)
pkl.dump(data, f)
for item in self.trackers:
data = L.get_all_param_values(item)
pkl.dump(data, f)
文章目录