Блог

Я работаю над обучением модели с помощью трансфертного обучения. Предварительно обученная модель, которую я использую, — это «SSD MobileNet V2 FPNLite 320×320» из модели Zoo TF2. И у меня есть некоторая путаница с пунктом train_config в файле pipeline.config .

Если я изменюсь num_steps , модель будет тренироваться вместе со num_steps мной . Но когда я меняюсь total_steps , модель все равно тренируется num_steps . Даже если я установлю num_steps > > total_step , ошибки не будет. И когда я проверяю все модели SSD в модели Zoo TF2, я всегда вижу, что total_steps то же num_steps самое, что и .

• Вопрос: Нужно ли мне устанавливать total_steps то же самое с num_steps помощью ? Какова связь между этим?

 train_config {
  batch_size: 128
  data_augmentation_options {
    random_horizontal_flip {
    }
  }
  data_augmentation_options {
    random_crop_image {
      min_object_covered: 0.0
      min_aspect_ratio: 0.75
      max_aspect_ratio: 3.0
      min_area: 0.75
      max_area: 1.0
      overlap_thresh: 0.0
    }
  }
  sync_replicas: true
  optimizer {
    momentum_optimizer {
      learning_rate {
        cosine_decay_learning_rate {
          learning_rate_base: 0.07999999821186066
          total_steps: 50000
          warmup_learning_rate: 0.026666000485420227
          warmup_steps: 1000
        }
      }
      momentum_optimizer_value: 0.8999999761581421
    }
    use_moving_average: false
  }
  fine_tune_checkpoint: "PATH_TO_BE_CONFIGURED"
  num_steps: 50000
  startup_delay_steps: 0.0
  replicas_to_aggregate: 8
  max_number_of_boxes: 100
  unpad_groundtruth_tensors: false
  fine_tune_checkpoint_type: "classification"
  fine_tune_checkpoint_version: V2
}