可以尝试通过转换为onnx的格式导入进来进行训练和推理。
ONNX是一种针对机器学习所设计的开放式的文件格式,用于存储训练好的模型。它使得不同的人工智能框架(如Pytorch, tensorflow等)可以采用相同格式存储模型数据并交互。
注意事项:
ONNX只是一个格式,就和json一样。只要你满足一定的规则,都算是合法的,因此单纯从Pytorch转成一个ONNX文件很简单,但是不同后端设备接受的onnx是不一样的。
所以某些模型转过来不一定能进行训练和推理,跟网络的模型结构有一定关系。且本身模型在转换成为onnx格式的时候,有些复杂的模型结构可能无法正确转换,具体模型最好是实测。