diff --git a/library/custom_train_functions.py b/library/custom_train_functions.py index fd4f6156..dde0bdd4 100644 --- a/library/custom_train_functions.py +++ b/library/custom_train_functions.py @@ -15,4 +15,4 @@ def apply_snr_weight(loss, timesteps, noise_scheduler, gamma): return loss def add_custom_train_arguments(parser: argparse.ArgumentParser): - parser.add_argument("--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.") + parser.add_argument("--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が推奨")