文件名称:Compare-PyTorch-Catalyst-Ignite-Lightning:通过示例代码比较PyTorch Catalyst,Ignite,Lightning
文件大小:8KB
文件格式:ZIP
更新时间:2024-05-22 13:39:55
Python
比较PyTorch催化剂,点燃,闪电 环境 Python 3.7 torch==1.3.1 torchvision==0.4.2 catalyst==19.12 pytorch-ignite==0.2.1 pytorch-lightning==0.5.3.2 如何使用 $ pip install -r requirements.txt # Default $ python train_default.py # Catalyst $ python train_catalyst.py # Ignite $ python train_ignite.py # Lightning $ python train_lightning.py
【文件预览】:
Compare-PyTorch-Catalyst-Ignite-Lightning-master
----.gitignore(2KB)
----README.md(407B)
----train_lightning.py(2KB)
----train_default.py(2KB)
----train_catalyst.py(1KB)
----share_funcs.py(1KB)
----output()
--------.gitkeep(0B)
----input()
--------.gitkeep(0B)
----requirements.txt(1KB)
----train_ignite.py(2KB)