@@ -18,13 +18,13 @@ class ScoreModel(pl.LightningModule):
...
@@ -18,13 +18,13 @@ class ScoreModel(pl.LightningModule):
defadd_argparse_args(parser):
defadd_argparse_args(parser):
parser.add_argument("--lr",type=float,default=1e-4,help="The learning rate (1e-4 by default)")
parser.add_argument("--lr",type=float,default=1e-4,help="The learning rate (1e-4 by default)")
parser.add_argument("--ema_decay",type=float,default=0.999,help="The parameter EMA decay constant (0.999 by default)")
parser.add_argument("--ema_decay",type=float,default=0.999,help="The parameter EMA decay constant (0.999 by default)")
parser.add_argument("--t_eps",type=float,default=0.03,help="The minimum time (3e-2 by default)")
parser.add_argument("--t_eps",type=float,default=0.03,help="The minimum process time (0.03 by default)")
parser.add_argument("--num_eval_files",type=int,default=20,help="Number of files for speech enhancement performance evaluation during training. Pass 0 to turn off (no checkpoints based on evaluation metrics will be generated).")
parser.add_argument("--num_eval_files",type=int,default=20,help="Number of files for speech enhancement performance evaluation during training. Pass 0 to turn off (no checkpoints based on evaluation metrics will be generated).")
parser.add_argument("--loss_type",type=str,default="mse",choices=("mse","mae"),help="The type of loss function to use.")
parser.add_argument("--loss_type",type=str,default="mse",choices=("mse","mae"),help="The type of loss function to use.")