【发布时间】:2020-02-07 01:08:42
【问题描述】:
我尝试使用https://github.com/tensorflow/models/tree/master/research/slim 中提到的 cifar10 数据集训练 mobilenetV2
DATASET_DIR=/tmp/cifar10
TRAIN_DIR=/tmp/train_logs
python train_image_classifier.py \
--train_dir=${TRAIN_DIR} \
--dataset_name=cifar10 \
--dataset_split_name=train \
--dataset_dir=${DATASET_DIR} \
--model_name=mobilenet_v2 \
--clone_cpu=True
【问题讨论】:
标签: tensorflow tensorflow-slim mobilenet