HRNet_W48_C.sh 302 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
#Training details
#HRNet_W48_C
python train.py \
       --model=HRNet_W48_C \
       --batch_size=256 \
       --total_images=1281167 \
       --class_dim=1000 \
       --lr_strategy=piecewise_decay \
       --lr=0.1 \
       --num_epochs=120 \
       --model_save_dir=output/ \
       --l2_decay=1e-4