Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- 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='
|
| 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 |
|