赞
踩
以DeepLabv3+为例进行讲解
# 设定batch_size的值即为单卡迭代一次送入网络的图片数量,一般显卡显存越大,batch_size的值可以越大
batch_size: 2
# epoch=(iters*batch_size*num_gpus)/N (x*2*4)/28990=80
iters: 289900 # 289900/(28990/(2*4))=80个epochs
# 训练数据设置
train_dataset:
# 选择数据集格式
type: Dataset
# 选择数据集路径
dataset_root: dataset/roadrisk
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。