文件名称：inferno:围绕PyTorch的实用程序库
文件大小：223KB
文件格式：ZIP
更新时间：2024-05-30 17:15:04
deep-learning pytorch neural-networks Python
地狱 Inferno是一个小库，提供围绕实用程序和便利功能/类。 这是一个进行中的工作，但v0.4以后的版本应该相当稳定！ 免费软件：Apache Software License 2.0 文档： : （正在进行中）。 特征 当前功能包括： 基本的用于封装训练样板（迭代/时期循环，验证和检查点创建）， 由提供支持的用于构建具有复杂架构的模型的 。 通过多个GPU， 用于torch.nn.Module级参数初始化， 用于数据预处理/转换， Tensorboard （最好安装atleast tensorflow-cpu ） 回调API，以实现与培训师的灵活交互， 各种公用事业层还在进行中， 体积数据集的子模块，还有更多！ import torch . nn as nn from inferno . io . box . cifar import get_cifar