@@ -1,12 +0,0 @@
|
||||
import torch
|
||||
import pandas as pd
|
||||
import json
|
||||
|
||||
def tensor_to_excel(tensor,output_path,file_name = "output"):
|
||||
df = pd.DataFrame(tensor.numpy())
|
||||
df.to_excel(output_path + '/' + file_name + '.xlsx', index=False)
|
||||
|
||||
def tensor_to_json(tensor,output_path,file_name = "output"):
|
||||
tensor_list = tensor.tolist() # 转换为Python列表
|
||||
with open(output_path + '/' + file_name +'.json', 'w') as f:
|
||||
json.dump(tensor_list, f) # 序列化为JSON
|
||||
@@ -1,22 +0,0 @@
|
||||
#1 pytorch 自有格式导出
|
||||
import torch
|
||||
|
||||
def export_model_pt(model,target,name = "model"):
|
||||
script_model = torch.jit.script(model) # 或 torch.jit.trace(model, input)
|
||||
script_model.save(target + name + ".pt")
|
||||
#2 通用格式导出
|
||||
def export_model_onnx(model,input_tensor,target,name="model"):
|
||||
torch.onnx.export(model, input_tensor, target+ name + ".onnx")
|
||||
#3二进制权重导出
|
||||
def export_model_bin(model,target,name = "weights"):
|
||||
torch.save(model.state_dict(), target + name + ".bin")
|
||||
|
||||
def export_model(model,target,file_name,name):
|
||||
if name == 'bin':
|
||||
return export_model_bin(model,target,file_name)
|
||||
if name == 'onnx':
|
||||
return export_model_onnx(model,target,file_name)
|
||||
if name == 'pt':
|
||||
return export_model_bin(model,target,file_name)
|
||||
else:
|
||||
raise ValueError(f"不支持的导出类型")
|
||||
@@ -1 +0,0 @@
|
||||
import FC_ML_Data
|
||||
Reference in New Issue
Block a user