文件名称:tfpyth:将TensorFlow放回PyTorch中,再放回TensorFlow中(不同的TensorFlow PyTorch适配器)
文件大小:7KB
文件格式:ZIP
更新时间:2024-02-25 07:15:04
machine-learning tensorflow pytorch MachinelearningPython
苯丙氨酸 将TensorFlow放回PyTorch中,再放回TensorFlow中(使用可区分的TensorFlow PyTorch适配器)。 您是否有一个使用TensorFlow的代码库和一个使用PyTorch的代码库,并且想要训练一个同时使用端到端的模型? 该库无需重写任何一个代码库就可以实现! 它允许您使用简单的函数包装TensorFlow图以使其可以通过PyTorch进行调用(和区分),反之亦然。 唯一的警告是张量必须通过CPU复制并路由,直到TensorFlow支持__cuda_array_interface (请在__cuda_array_interface星)。 安装
【文件预览】:
tfpyth-master
----setup.py(3KB)
----.gitignore(50B)
----pyproject.toml(228B)
----.travis.yml(229B)
----LICENSE(1KB)
----setup.cfg(55B)
----examples()
--------torch_from_tensorflow.py(553B)
----README.md(2KB)
----tests()
--------test_adapters.py(2KB)
----.coveragerc(55B)
----tfpyth()
--------__init__.py(4KB)