tensor和numpy不是一种数据类型,所以,在将数据输出保存之前,需要将tensor的数据类型进行转换
# var保存成文件(numpy array)
var_path = os.path.join(save_model_path, "Enpy%i.npy" % epoch)
np_var = var.data.cpu().numpy() # 数据类型转换
print(
tensor和numpy不是一种数据类型,所以,在将数据输出保存之前,需要将tensor的数据类型进行转换
# var保存成文件(numpy array)
var_path = os.path.join(save_model_path, "Enpy%i.npy" % epoch)
np_var = var.data.cpu().numpy() # 数据类型转换
print(