Skip to content

Commit 5a852b5

Browse files
authored
Merge pull request locuslab#2 from andrewwilson/master
Minor correction to the advertised default values of command line arguments.
2 parents 861465c + 649ceed commit 5a852b5

File tree

4 files changed

+13
-13
lines changed

4 files changed

+13
-13
lines changed

TCN/adding_problem/add_test.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,13 @@
1717
parser.add_argument('--epochs', type=int, default=10,
1818
help='upper epoch limit (default: 10)')
1919
parser.add_argument('--ksize', type=int, default=7,
20-
help='kernel size (default: 8)')
20+
help='kernel size (default: 7)')
2121
parser.add_argument('--levels', type=int, default=8,
22-
help='# of levels (default: 5)')
22+
help='# of levels (default: 8)')
2323
parser.add_argument('--seq_len', type=int, default=400,
2424
help='sequence length (default: 400)')
2525
parser.add_argument('--log-interval', type=int, default=100, metavar='N',
26-
help='report interval (default: 10')
26+
help='report interval (default: 100')
2727
parser.add_argument('--lr', type=float, default=4e-3,
2828
help='initial learning rate (default: 4e-3)')
2929
parser.add_argument('--optim', type=str, default='Adam',

TCN/char_cnn/char_cnn_test.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,15 @@
2121
parser.add_argument('--dropout', type=float, default=0.1,
2222
help='dropout applied to layers (default: 0.1)')
2323
parser.add_argument('--emb_dropout', type=float, default=0.1,
24-
help='dropout applied to the embedded layer (0 = no dropout)')
24+
help='dropout applied to the embedded layer (0 = no dropout) (default: 0.1)')
2525
parser.add_argument('--clip', type=float, default=0.15,
2626
help='gradient clip, -1 means no clip (default: 0.15)')
2727
parser.add_argument('--epochs', type=int, default=100,
2828
help='upper epoch limit (default: 100)')
2929
parser.add_argument('--ksize', type=int, default=3,
30-
help='kernel size (default: 5)')
30+
help='kernel size (default: 3)')
3131
parser.add_argument('--levels', type=int, default=3,
32-
help='# of levels (default: 4)')
32+
help='# of levels (default: 3)')
3333
parser.add_argument('--log-interval', type=int, default=100, metavar='N',
3434
help='report interval (default: 100')
3535
parser.add_argument('--lr', type=float, default=4,
@@ -39,7 +39,7 @@
3939
parser.add_argument('--optim', type=str, default='SGD',
4040
help='optimizer to use (default: SGD)')
4141
parser.add_argument('--nhid', type=int, default=450,
42-
help='number of hidden units per layer (default: 150)')
42+
help='number of hidden units per layer (default: 450)')
4343
parser.add_argument('--validseqlen', type=int, default=320,
4444
help='valid sequence length (default: 320)')
4545
parser.add_argument('--seq_len', type=int, default=400,

TCN/copy_memory/copymem_test.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,13 @@
2525
parser.add_argument('--iters', type=int, default=100,
2626
help='number of iters per epoch (default: 100)')
2727
parser.add_argument('--levels', type=int, default=8,
28-
help='# of levels (default: 5)')
28+
help='# of levels (default: 8)')
2929
parser.add_argument('--blank_len', type=int, default=1000, metavar='N',
3030
help='The size of the blank (i.e. T) (default: 1000)')
3131
parser.add_argument('--seq_len', type=int, default=10,
3232
help='initial history size (default: 10)')
3333
parser.add_argument('--log-interval', type=int, default=50, metavar='N',
34-
help='report interval (default: 10')
34+
help='report interval (default: 50')
3535
parser.add_argument('--lr', type=float, default=5e-4,
3636
help='initial learning rate (default: 5e-4)')
3737
parser.add_argument('--optim', type=str, default='RMSprop',

TCN/mnist_pixel/pmnist_test.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -13,19 +13,19 @@
1313
parser.add_argument('--cuda', action='store_false',
1414
help='use CUDA (default: True)')
1515
parser.add_argument('--dropout', type=float, default=0.05,
16-
help='dropout applied to layers (default: 0.0)')
16+
help='dropout applied to layers (default: 0.05)')
1717
parser.add_argument('--clip', type=float, default=-1,
1818
help='gradient clip, -1 means no clip (default: -1)')
1919
parser.add_argument('--epochs', type=int, default=20,
2020
help='upper epoch limit (default: 20)')
2121
parser.add_argument('--ksize', type=int, default=7,
22-
help='kernel size (default: 8)')
22+
help='kernel size (default: 7)')
2323
parser.add_argument('--levels', type=int, default=8,
2424
help='# of levels (default: 8)')
2525
parser.add_argument('--log-interval', type=int, default=100, metavar='N',
26-
help='report interval (default: 10')
26+
help='report interval (default: 100')
2727
parser.add_argument('--lr', type=float, default=2e-3,
28-
help='initial learning rate (default: 5e-4)')
28+
help='initial learning rate (default: 2e-3)')
2929
parser.add_argument('--optim', type=str, default='Adam',
3030
help='optimizer to use (default: Adam)')
3131
parser.add_argument('--nhid', type=int, default=25,

0 commit comments

Comments
 (0)