Skip to content

Commit 5e5f189

Browse files
authored
Update run.py
1 parent 6dff74a commit 5e5f189

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

run.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949

5050
# model define
5151
parser.add_argument('--top_k', type=int, default=5, help='for TimesBlock')
52-
parser.add_argument('--num_kernels', type=int, default=6, help='for Inception')
52+
parser.add_argument('--num_kernels', type=int, default=3, help='for Inception')
5353
parser.add_argument('--enc_in', type=int, default=7, help='encoder input size')
5454
parser.add_argument('--dec_in', type=int, default=7, help='decoder input size')
5555
parser.add_argument('--c_out', type=int, default=7, help='output size')

0 commit comments

Comments
 (0)