没有合适的资源?快使用搜索试试~ 我知道了~
PyTorch-Project-Template
共36个文件
py:18个
yaml:9个
sh:5个
需积分: 5 0 下载量 56 浏览量
2021-04-07
12:27:57
上传
评论
收藏 26KB ZIP 举报
温馨提示
PyTorch项目模板 安装 $ git clone git@github.com:tomoino/PyTorch-Project-Template.git 用法 设置 $ cd PyTorch-Project-Template $ sh docker/build.sh $ sh docker/run.sh $ sh docker/exec.sh 开始一个新项目 将yaml文件添加到“ ./configs/project” $ vi ./configs/project/new_project.yaml 使用-cn(或--config-name)标志运行train.py以指定项目$ python train.py -cn new_project 训练 $ python train.py 网格搜索 您可以使用多种不同的配置运行train.py。 $ python train.py -m \
资源推荐
资源详情
资源评论
收起资源包目录
PyTorch-Project-Template-main.zip (36个子文件)
PyTorch-Project-Template-main
docker
init.sh 226B
build.sh 204B
exec.sh 120B
Dockerfile 2KB
run.sh 299B
requirements.txt 108B
env.sh 252B
train.py 626B
models
helper.py 1KB
__init__.py 780B
base_model.py 3KB
networks
simple_cnn.py 1KB
resnet18.py 656B
configs
project
hydra
job_logging
custom.yaml 395B
train
trainer
default.yaml 53B
criterion
cross_entropy.yaml 71B
metric
classification.yaml 72B
optimizer
adam.yaml 74B
default.yaml 545B
model
simple_cnn.yaml 68B
data
dataset
cifar10.yaml 225B
sampler
balanced_batch_sampler.yaml 75B
supported_info.py 486B
metrics
classification_metric.py 3KB
__init__.py 739B
trainers
base_trainer.py 2KB
__init__.py 653B
default_trainer.py 4KB
README.md 3KB
data
helper.py 2KB
dataloader.py 614B
__init__.py 4KB
dataset
omniglot.py 928B
cifar10.py 1KB
sampler
balanced_batch_sampler.py 2KB
.gitignore 2KB
共 36 条
- 1
资源评论
彭仕安
- 粉丝: 24
- 资源: 4679
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功