CPU环境 下训练模型(image_classification)有Bug
Created by: OliverLPH
在图像识别的路径中，尝试在CPU下跑模型，均有该bug
python train.py \
> --model=VGG16 \
> --batch_size=256 \
> --total_images=1281167 \
> --class_dim=1000 \
> --lr_strategy=cosine_decay \
> --image_shape=3,224,224 \
> --model_category=models_name \
> --model_save_dir=output/ \
> --lr=0.01 \
> --num_epochs=90 \
> --with_mem_opt=True \
> --l2_decay=3e-4 \
> --use_gpu=False参数配置如下图，关掉了gpu
报错如下
