翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。
イメージ分類ハイパーパラメータ
ハイパーパラメータは、機械学習モデルが学習を開始する前に設定されるパラメータです。以下のハイパーパラメータは、Amazon の SageMaker 組み込み画像分類アルゴリズムでサポートされています。イメージ分類ハイパーパラメータの調整については、「イメージ分類モデルを調整する」を参照してください。
Parameter Name | 説明 |
---|---|
num_classes |
出力クラスの数。このパラメータは、ネットワーク出力のディメンションを定義し、通常はデータセット内のクラス数に設定されます。 マルチクラス分類に加えて、マルチラベル分類もサポートされています。拡張マニフェストファイルを使用したマルチラベル分類の操作方法の詳細については、「イメージ分類アルゴリズムの入出力インターフェイス」を参照してください。 必須 有効な値: 正の整数 |
num_training_samples |
入力データセット内のトレーニング例の数。 この値とトレーニングセット内のサンプル数が一致しない場合、 必須 有効な値: 正の整数 |
augmentation_type |
データ補強タイプ。入力イメージは、次に示す複数の方法で補強できます。
オプション 有効な値: デフォルト値: なし |
beta_1 |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.9 |
beta_2 |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.999 |
checkpoint_frequency |
モデルパラメータを保存する期間 (エポック数)。 すべてのチェックポイントファイルは、最終的なモデルファイル「model.tar.gz」の一部として保存され、指定されたモデルの場所に S3 にアップロードされることに注意してください。これにより、トレーニング中に保存されたチェックポイントの数に比例してモデルファイルのサイズが大きくなります。 オプション 有効な値: デフォルト値: なし (検証精度が最も高いエポックにチェックポイントを保存します) |
early_stopping |
トレーニング中に早期停止ロジックを使用する場合は オプション 有効な値: デフォルト値: |
early_stopping_min_epochs |
早期停止ロジックを呼び出す前に実行する必要があるエポックの最小数。 オプション 有効な値: 正の整数 デフォルト値: 10 |
early_stopping_patience |
関連メトリクスで改善が見られなかった場合にトレーニングを終了する前に待機するエポックの数。 オプション 有効な値: 正の整数 デフォルト値: 5 |
early_stopping_tolerance |
精度検証メトリクスの改善を測定する相対的な許容値。精度の改善を前回の最高精度で除算した比率が オプション 有効な値: 0 ≤ 浮動小数点数 ≤ 1 デフォルト値: 0.0 |
epochs |
トレーニングエポックの数。 オプション 有効な値: 正の整数 デフォルト値: 30 |
eps |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 1e-8 |
gamma |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.9 |
image_shape |
入力イメージのディメンション。これはネットワークの入力レイヤーと同じサイズになります。フォーマットは " トレーニングの場合、任意の次元でいずれかの入力イメージがこのパラメータよりも小さいと、トレーニングは失敗します。イメージが大きい場合、このパラメータで指定されたクロップ領域を使用して、イメージの一部がクロップされます。ハイパーパラメータ 推論時に、入力イメージのサイズはトレーニング中に使用された オプション 有効な値: 文字列 デフォルト値: '3,224,224' |
kv_store |
分散型トレーニング中の重み更新同期モード。重み更新は、複数マシン間で同期的または非同期的に更新できます。同期更新は、一般には非同期更新よりも精度が高くなりますが、低速な可能性があります。詳細については、「」の「分散トレーニングMXNet」を参照してください。 このパラメータは、単一のマシントレーニングには適用されません。
オプション 有効な値: デフォルト値: なし |
learning_rate |
初期学習レート。 オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.1 |
lr_scheduler_factor |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.1 |
lr_scheduler_step |
学習レートを下げるエポック。 オプション 有効な値: 文字列 デフォルト値: なし |
mini_batch_size |
トレーニングのバッチサイズ。単一マシンのマルチGPU設定では、各 が オプション 有効な値: 正の整数 デフォルト値: 32 |
momentum |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.9 |
multi_label |
各サンプルに複数のラベルを割り当てることができる複数ラベル分類に使用するフラグ。全クラスの平均精度が記録されます。 オプション 有効な値: 0 または 1 デフォルト値: 0 |
num_layers |
ネットワークのレイヤー数。画像サイズが大きいデータ (224x224 など ImageNet) の場合、セット [18、34、50、101、152、200] からレイヤー数を選択することをお勧めします。イメージサイズが小さいデータ (例: 28x28 - などCIFAR) の場合は、セット [20、32、44、56、110] からレイヤーの数を選択することをお勧めします。各セットのレイヤーの数は、 ResNet 紙に基づいています。転移学習では、レイヤー数は基本ネットワークのアーキテクチャを定義するため、セット [18、34、50、101、152、200] からのみ選択できます。 オプション 有効な値: [18, 34, 50, 101, 152, 200] または [20, 32, 44, 56, 110] 内の正の整数。 デフォルト値: 152 |
optimizer |
オプティマイザのタイプ。オプティマイザのパラメータの詳細については、「」のMXNet「」を参照してくださいAPI。 オプション 有効な値:
デフォルト値: |
precision_dtype |
トレーニングに使用される重みの精度。アルゴリズムは、重みに単精度 ( オプション 有効な値: デフォルト値: |
resize |
トレーニングのためにイメージのサイズを変更した後、イメージの最短辺にあるピクセル数です。パラメータが設定されていない場合、トレーニングデータはサイズ変更なしで使用されます。パラメータは、トレーニングの失敗を防ぐため、 イメージコンテンツタイプを使用する場合は必須 RecordIO コンテンツタイプを使用する場合はオプション 有効な値: 正の整数 デフォルト値: なし |
top_k |
トレーニング中に top-k 精度を報告します。top-1 トレーニング精度はすでに報告されている通常のトレーニング精度と同じであるため、このパラメータは 1 より大きい必要があります。 オプション 有効な値: 1 より大きい正の整数。 デフォルト値: なし |
use_pretrained_model |
トレーニングに事前トレーニング済みモデルを使用するためのフラグ。1 に設定すると、対応するレイヤー数の事前トレーニング済みモデルがロードされ、トレーニングに使用されます。最上位 FC レイヤーのみがランダムな重みで再初期化されます。それ以外の場合、ネットワークはゼロからトレーニングされます。 オプション 有効な値: 0 または 1 デフォルト値: 0 |
use_weighted_loss |
重みがクラスの分布に基づいて計算される、複数ラベル分類に重み付き交差エントロピー損失を使用するためのフラグ ( オプション 有効な値: 0 または 1 デフォルト値: 0 |
weight_decay |
オプション 有効な値: 浮動小数点数。[0, 1] の範囲です。 デフォルト値: 0.0001 |