|
31 | 31 | help='Initial learning rate.')
|
32 | 32 | parser.add_argument('--weight_decay', type=float, default=0.0,
|
33 | 33 | 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, |
35 | 35 | 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, |
37 | 37 | help='Number of hidden units.')
|
38 |
| -parser.add_argument('--hid2', type=int, default=6, |
| 38 | +parser.add_argument('--hid2', type=int, default=25, |
39 | 39 | 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, |
41 | 41 | help='Smooth for Res layer')
|
42 | 42 | parser.add_argument('--dropout', type=float, default=0.9,
|
43 | 43 | help='Dropout rate (1 - keep probability).')
|
|
0 commit comments