未提供项目说明
deep-learning-utils的Python项目详细描述
dl-utils:PyTorch深度学习实用程序
内容
该包主要包含函数和模块化以及(只要可能)纯PyTorch实现中的损失函数、模型定义和度量。在
安装
来源
pip install git+https://github.com/justusschock/dl-utils
从PyPi
pip install deep-learning-utils
分装
目前有以下子包:
dlutils.data
:包含数据实用程序(到目前为止只是一个随机伪数据的数据集)dlutils.losses
:将PyTorch本身给出的损失扩展为几个损失函数dlutils.metrics
:实现一些常见的度量dlutils.models
:包含许多流行模型的Nd实现dlutils.models.gans
:包含许多基本的gan实现,但到目前为止还没有用于任意维度
dlutils.optims
:containis附加优化器dlutils.utils
:包含张量操作和模块加载等附加实用程序
注意
- 在
这些代码中的大部分只进行了少量的测试,没有使用适当的CI/CD和unittests。我目前正在做这方面的工作,任何贡献都是非常欢迎的。在
在 - 在
所有实现都是针对纯PyTorch完成的。您可以在您想要的任何培训框架中使用它们(比如[pytorch/ignite]{a1})或Pytorch-Lightning),也可以在您的定制培训循环中使用它们
在
- 项目
标签: