diff --git a/opt/opt.py b/opt/opt.py index 5b37459c..4e454870 100644 --- a/opt/opt.py +++ b/opt/opt.py @@ -119,7 +119,7 @@ group = parser.add_argument_group("initialization") group.add_argument('--init_sigma', type=float, - default=0.1,#0.1, + default=0.1, help='initialization sigma') group = parser.add_argument_group("misc experiments") @@ -138,7 +138,7 @@ group.add_argument('--tune_mode', action='store_true', default=False, help='hypertuning mode (do not save, for speed)') -group.add_argument('--rms_beta', type=float, default=0.99) +group.add_argument('--rms_beta', type=float, default=0.95) group.add_argument('--lambda_tv', type=float, default=1e-5) group.add_argument('--tv_sparsity', type=float, default=0.01) group.add_argument('--tv_logalpha', action='store_true', default=False,