alwaysgood commited on
Commit
558291d
·
verified ·
1 Parent(s): da082ac

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +3 -1
inference.py CHANGED
@@ -7,6 +7,8 @@ import os
7
  from tqdm import tqdm
8
  import json
9
  import sys
 
 
10
  sys.path.append('.')
11
 
12
  from models import TimeXer
@@ -42,7 +44,7 @@ parser.add_argument('--embed', type=str, default='timeF', help='time features en
42
  parser.add_argument('--activation', type=str, default='gelu', help='activation')
43
  parser.add_argument('--output_attention', action='store_true', help='whether to output attention in ecoder')
44
  parser.add_argument('--use_norm', type=int, default=1, help='whether to use normalize')
45
- parser.add_argument('--freq', type=str, default='t', help='freq for time features encoding')
46
  args = parser.parse_args()
47
 
48
 
 
7
  from tqdm import tqdm
8
  import json
9
  import sys
10
+ import traceback
11
+
12
  sys.path.append('.')
13
 
14
  from models import TimeXer
 
44
  parser.add_argument('--activation', type=str, default='gelu', help='activation')
45
  parser.add_argument('--output_attention', action='store_true', help='whether to output attention in ecoder')
46
  parser.add_argument('--use_norm', type=int, default=1, help='whether to use normalize')
47
+ parser.add_argument('--freq', type=str, default='T', help='freq for time features encoding')
48
  args = parser.parse_args()
49
 
50