跳转到主要内容

pytorch训练工具

项目描述

basenet

封装基本pytorch功能的类。

  • 训练一个epoch
  • 评估一个epoch
  • 预测
  • 学习率调度
安装
conda create -n basenet4_env python=3.6 pip -y
source activate basenet4_env

pip install -r requirements.txt
conda install -y pytorch torchvision cuda90 -c pytorch
pip install -e .

示例

cd examples/cifar
./run.sh

项目详情


下载文件

下载适用于您平台文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分发

basenet-0.1.0.tar.gz (10.0 kB 查看哈希)

上传时间: 源代码

由以下机构支持

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误记录 StatusPage StatusPage 状态页面