训练中同时使用tensorflow与pytorch
在开源的 TfPyTh 中,不论是 TensorFlow 还是 PyTorch 计算图,它们都可以包装成一个可微函数,并在另一个框架中高效完成前向与反向传播。
github项目地址:BlackHC/TfPyTh。
神经网络交换格式 ONNX,定义了一种通用的计算图。目前 ONNX 已经原生支持 MXNet、PyTorch 和 Caffe2 等大多数框架,但是像 TensorFlow 或 Keras 之类的只能通过第三方转换器转换为 ONNX 格式。TfPyTh无需改写已有的代码就能在框架间自由转换。T.
转载
2021-04-11 12:09:41 ·
2562 阅读 ·
0 评论