跳转到主要内容

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

由以下机构支持

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