Какие переменные нужно тренировать при обучении последнему слою inception_v4.(Trainable_scopes / checkpoint_exclude_scopes) - PullRequest
0 голосов
/ 25 января 2019

Я использовал следующую команду для переобучения модели inception_resnet_v2.

tcmd = "python train_image_classifier.py \
  --train_dir="+TRAIN_DIR+" \
  --dataset_name=training_images \
  --dataset_split_name=train \
  --dataset_dir="+DATASET_DIR+" \
  --model_name=inception_resnet_v2 \
  --checkpoint_path="+PRETRAINED_CHECKPOINT_DIR+"/inception_resnet_v2_2016_08_30.ckpt \
  --checkpoint_exclude_scopes=InceptionResnetV2/Logits,InceptionResnetV2/AuxLogits  \
  --trainable_scopes=InceptionResnetV2/Logits,InceptionResnetV2/AuxLogits  \
  --max_number_of_steps=1000 \
  --batch_size=16 \
  --learning_rate=0.01 \
  --learning_rate_decay_type=fixed \
  --save_interval_secs=60 \
  --save_summaries_secs=60 \
  --log_every_n_steps=100 \
  --optimizer=rmsprop \
  --weight_decay=0.00004"

os.system(tcmd)

Теперь я хочу изменить команду для обучения inception_v4. Я уже скачал контрольную точку для этой модели, но я не могу понять, каковы trainable_scopes или checkpoint_exclude_scopes для переподготовки последнего слоя этой модели.

Может кто-нибудь, пожалуйста, скажите мне, что это такое.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...