【文件属性】:
文件名称:argus:轻量级库,用于在PyTorch中训练神经网络
文件大小:316KB
文件格式:ZIP
更新时间:2021-05-04 12:36:12
python machine-learning deep-learning neural-network pytorch
Argus是一个轻量级的库,用于在PyTorch中训练神经网络。
文献资料
安装
要求:
火炬> = 1.1.0
从点子:
pip install pytorch-argus
来自来源:
pip install -U git+https://github.com/lRomul/argus.git
例子
与简单的图像分类例如create_model从:
from torchvision . datasets import MNIST
from torch . utils . data import DataLoader
from torchvision . transforms import Compose , ToTensor , Normalize
import timm
import argus
from argus . callbacks import MonitorCh
【文件预览】:
argus-master
----Dockerfile(1KB)
----.gitignore(1KB)
----setup.cfg(196B)
----README.md(4KB)
----.github()
--------workflows()
----tests()
--------metrics()
--------conftest.py(5KB)
--------callbacks()
--------test_pipeline.py(2KB)
--------test_utils.py(4KB)
--------model()
--------__init__.py(0B)
--------requirements.txt(59B)
--------test_optimizer.py(885B)
--------engine()
--------test_loss.py(508B)
----docs()
--------make.bat(799B)
--------source()
--------requirements.txt(41B)
--------Makefile(638B)
----examples()
--------cifar_advanced.py(8KB)
--------custom_events.py(2KB)
--------README.md(2KB)
--------load_model.py(2KB)
--------cifar_simple.py(3KB)
--------cifar_advanced.sh(115B)
--------mnist.py(4KB)
--------requirements.txt(47B)
--------custom_build_methods.py(2KB)
--------quickstart.ipynb(9KB)
--------build_models.ipynb(5KB)
--------mnist_vae.py(5KB)
----LICENSE(1KB)
----requirements.txt(13B)
----MANIFEST.in(41B)
----setup.py(2KB)
----Makefile(624B)
----assets()
--------logo()
----argus()
--------optimizer.py(644B)
--------metrics()
--------callbacks()
--------types.py(343B)
--------model()
--------utils.py(6KB)
--------__init__.py(148B)
--------loss.py(587B)
--------engine()