文件名称：pytorch-tools:PyTorch的工具箱
文件大小：797KB
文件格式：ZIP
更新时间：2024-03-29 21:37:13
Python
火炬工具 用于快速原型制作的PyTorch工具箱。 概述 - 像模型教练一样 -收集不同的损失函数。 -集合。 -分类模型动物园。 -细分模型动物园 -用于轻松增加测试时间的包装器 安装 需要已安装的GPU驱动程序和CUDA。 pip install git+https://github.com/bonlime/pytorch-tools.git@master 还建议安装NVIDIA Apex，以允许使用其他优化器 pip install ---upgrade -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git