image_classification 训练自定义数据集batch问题
Created by: zouxiaodong
数据集总共:6390,训练集有 5112行记录,batch_size设置为8,训练脚本输出的batch,train 和 test 只到70+
执行的脚本如下:
python train.py
--data_dir=/root/paddle/lemon/data/
--model=ResNet50
--batch_size=8
--total_images=6390
--class_dim=3
--image_shape=3,900,900
--model_save_dir=/root/paddle/lemon/train_model/
--lr_strategy=piecewise_decay
--lr=0.1
--reader_thread=8
--use_gpu=false