Update help arguments for soft_min_snr_gamma

This commit is contained in:
rockerBOO
2024-04-01 16:10:19 -04:00
parent 38ef8ea8d6
commit d004690e24

View File

@@ -117,7 +117,7 @@ def add_custom_train_arguments(parser: argparse.ArgumentParser, support_weighted
"--soft_min_snr_gamma",
type=float,
default=None,
help="gamma for reducing the weight of high loss timesteps. Lower numbers have stronger effect. 5 is recommended by paper. / 低いタイムステップでの高いlossに対して重みを減らすためのgamma値、低いほど効果が強く、論文では5が推奨",
help="gamma for reducing the weight of high loss timesteps. Lower numbers have stronger effect. 1 is recommended. / 低いタイムステップでの高いlossに対して重みを減らすためのgamma値、低いほど効果が強く、論文では1が推奨",
)
parser.add_argument(
"--scale_v_pred_loss_like_noise_pred",