diff --git a/library/custom_train_functions.py b/library/custom_train_functions.py index 6bd01a06..32d2ede6 100644 --- a/library/custom_train_functions.py +++ b/library/custom_train_functions.py @@ -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",