Я использовал следующую команду для переобучения модели 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
для переподготовки последнего слоя этой модели.
Может кто-нибудь, пожалуйста, скажите мне, что это такое.