Skip to content

Commit 9b9ae5f

Browse files
authored
Update train.py
1 parent a5d8c5d commit 9b9ae5f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

train.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,13 @@
3131
help='Initial learning rate.')
3232
parser.add_argument('--weight_decay', type=float, default=0.0,
3333
help='Weight decay (L2 loss on parameters).')
34-
parser.add_argument('--l1', type=float, default=0.0,
34+
parser.add_argument('--l1', type=float, default=0.05,
3535
help='Weight decay (L1 loss on parameters).')
36-
parser.add_argument('--hid1', type=int, default=11,
36+
parser.add_argument('--hid1', type=int, default=13,
3737
help='Number of hidden units.')
38-
parser.add_argument('--hid2', type=int, default=6,
38+
parser.add_argument('--hid2', type=int, default=25,
3939
help='Number of hidden units.')
40-
parser.add_argument('--smoo', type=float, default=0.35,
40+
parser.add_argument('--smoo', type=float, default=0.5,
4141
help='Smooth for Res layer')
4242
parser.add_argument('--dropout', type=float, default=0.9,
4343
help='Dropout rate (1 - keep probability).')

0 commit comments

Comments
 (0)