在训练inception\u v4的最后一层时,需要训练哪些变量。(可训练\u范围/检查点\u排除\u范围)

2024-05-28 23:00:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用了以下命令来重新训练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 \
  --weight_decay=0.00004"

os.system(tcmd)

现在我想更改命令来训练inception_v4。我已经下载了这个模型的检查点,但是我不知道重新训练这个模型最后一层的trainable_scopescheckpoint_exclude_scopes是什么。你知道吗

有人能告诉我那些是什么吗。你知道吗


Tags: name模型命令dirtraindatasetexcludev2

热门问题