We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 54c5345 commit 2817a54Copy full SHA for 2817a54
1 file changed
train.py
@@ -44,8 +44,6 @@ def get_args():
44
parser.add_argument('--optim', type=str, default='adamw', help='select optimizer for training, '
45
'suggest using \'admaw\' until the'
46
' very final stage then switch to \'sgd\'')
47
- parser.add_argument('--alpha', type=float, default=0.25)
48
- parser.add_argument('--gamma', type=float, default=1.5)
49
parser.add_argument('--num_epochs', type=int, default=500)
50
parser.add_argument('--val_interval', type=int, default=1, help='Number of epoches between valing phases')
51
parser.add_argument('--save_interval', type=int, default=500, help='Number of steps between saving')
0 commit comments