DenseNet169.sh 325 字节
Newer Older
R
ruri 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
#Training details
#DenseNet169
python train.py \
      --model=DenseNet169 \
      --batch_size=256 \
      --total_images=1281167 \
      --image_shape=3,224,224 \
      --class_dim=1000 \
      --lr_strategy=piecewise_decay \
      --lr=0.1 \
      --num_epochs=120 \
      --model_save_dir=output/ \
      --l2_decay=1e-4