1 Star 0 Fork 1

luoqi/wide-resnet.pytorch

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
config.py 787 Bytes
一键复制 编辑 原始数据 按行查看 历史
bumsoo 提交于 2017-06-08 13:16 . Commit #7: Configuration file
############### Pytorch CIFAR configuration file ###############
import math
start_epoch = 1
num_epochs = 200
batch_size = 128
optim_type = 'SGD'
mean = {
'cifar10': (0.4914, 0.4822, 0.4465),
'cifar100': (0.5071, 0.4867, 0.4408),
}
std = {
'cifar10': (0.2023, 0.1994, 0.2010),
'cifar100': (0.2675, 0.2565, 0.2761),
}
# Only for cifar-10
classes = ('plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
def learning_rate(init, epoch):
optim_factor = 0
if(epoch > 160):
optim_factor = 3
elif(epoch > 120):
optim_factor = 2
elif(epoch > 60):
optim_factor = 1
return init*math.pow(0.2, optim_factor)
def get_hms(seconds):
m, s = divmod(seconds, 60)
h, m = divmod(m, 60)
return h, m, s
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/qiluo22/wide-resnet.pytorch.git
git@gitee.com:qiluo22/wide-resnet.pytorch.git
qiluo22
wide-resnet.pytorch
wide-resnet.pytorch
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385