-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathFC_main.py
144 lines (127 loc) · 7.02 KB
/
FC_main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
import os
import argparse
import logging
import numpy as np
import torch
from torch.backends import cudnn
from Forecasting.FC_solver import Solver
def mkdir(directory):
if not os.path.exists(directory):
os.makedirs(directory)
def str2bool(v):
return v.lower() in ('true')
def main(config):
cudnn.benchmark = True
if (not os.path.exists(config.model_save_path)):
mkdir(config.model_save_path)
# config.scheduler = 1
# solver = Solver(vars(config))
# solver.train()
# del solver
# config.scheduler = 0
solver = Solver(vars(config))
if config.mode == "train":
solver.train()
# elif config.mode == 'test':
solver.test()
elif config.mode == "test":
solver.test()
elif config.mode == 'ct':
solver.complexity_test()
return solver
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--log_path", type=str, default='./Forecasting/Logs_/logs_', help="path to save all the products from each trainging")
parser.add_argument("--id_", type=int, default=0, help="Run id")
parser.add_argument("--data_path", type=str, default='./data', help="path to grab data")
parser.add_argument("--description", type=str, default='', help="optional desciption")
parser.add_argument("--dataset", type=str, default="ETTm1", choices=["ETTh1", "ETTh2", "ETTm1", "ETTm2", "electricity",
"traffic", "weather"])
parser.add_argument('--mode', type=str, default='train', choices=['train', 'test', 'ct'])
# Save path
parser.add_argument('--model_save_path', type=str, default='checkpoints')
parser.add_argument('--plots_save_path', type=str, default='plots')
parser.add_argument('--his_save_path', type=str, default='hist')
# Training params
parser.add_argument("--seed", type=int, default=55)
parser.add_argument("--gpu_dev", type=str, default="6")
parser.add_argument("--n_epochs", type=int, default=50)
parser.add_argument("--batch", type=int, default=32)
parser.add_argument("--batch_testing", type=int, default=32)
parser.add_argument("--patience", type=int, default=10)
# optimizer
parser.add_argument("--lr_", type=float, default=5e-4, help= "learning rate")
parser.add_argument("--loss_type", type=str, default="TFDR", help= "TD: time domain loss, FD: Frequency domain loss, TFD: Mixture, TFDR")
parser.add_argument("--lamda", type=float, default=0.5, help = "lower puts more contriubtion from frequency loss")
parser.add_argument("--scheduler", type=int, default=0)
### Scheduler params
parser.add_argument("--warm_up", type=float, default=0.2, help="portion of warm up given number of epoches, e.g., 20 percent by defualt")
parser.add_argument("--start_lr", type=float, default=1e-5, help="starting learning rate")
parser.add_argument("--ref_lr", type=float, default=1.5e-4, help= "peak learning rate")
parser.add_argument("--final_lr", type=float, default=1e-5, help = "final learning rate")
parser.add_argument("--start_wd", type=float, default=0., help = "starting weight decay, setting it to 0. means no decay and decay scheduler")
parser.add_argument("--final_wd", type=float, default=0., help = "fianl weight decay")
# NFM params
parser.add_argument("--std", type=float, default=0.06, help="std factor used for initialization")
parser.add_argument('--input_c', type=int, default=7)
parser.add_argument("--hidden_dim", type=int, default=64, help = "dimension of hidden feature d")
parser.add_argument("--inff_siren_hidden", type=int, default=64)
parser.add_argument("--inff_siren_omega", type=float, default=30)
parser.add_argument("--hidden_factor", type=int, default=3)
parser.add_argument("--layer_num", type=int, default=1)
parser.add_argument("--dropout", type=float, default=0.3)
parser.add_argument("--filter_type", type=str, default="INFF", choices=["INFF", "FNO", "AFNO", "GFN", "AFF"])
parser.add_argument("--ff_projection_ex", type=int, default=3)
# LFT (based on siren) params
parser.add_argument("--lft", type=int, default=1)
parser.add_argument("--siren_hidden", type=int, default=48)
parser.add_argument("--siren_in_dim", type=int, default=4, help = "This is also parameter for INFF")
parser.add_argument("--siren_omega", type=float, default=30.)
parser.add_argument("--ff_std", type=int, default=128)
parser.add_argument("--lft_norm", type=int, default=0, help = "Whether to apply normalization to input spectrum in LFT")
parser.add_argument("--tau", type=str, default="independent", choices= ["independent", "shared"])
# Forecasting params
parser.add_argument("--norm_trick", type=str, default= "mean_std", choices= ["mean", "mean_std"],
help = "Statistics to use in input normalization and output denormalization")
parser.add_argument("--freq_span", type=int, default=-1, help = "-1 for full span modelling")
parser.add_argument("--channel_dependence", type=int, default=1, help = "channel_dependence -- 1: True, 0: False (channel independent)")
parser.add_argument("--look_back", type=int, default=720)
parser.add_argument("--horizon", type=int, default=96)
# IN-Out for training
parser.add_argument(
"--vars_in_train",
nargs='+',
type=int,
default=[360, 360, 96, 96],
help="A set of variables [Fs, F_in, L_out (horizon), L_in (lookback)] for formatting training data")
# IN-Out for testing
parser.add_argument(
"--vars_in_test",
nargs='+',
type=int,
default=[360, 360, 180, 360],
help="A set of variables [Fs, F_in, L_out (horizon), L_in (lookback)] \
for formatting testing data. If same as 'vars_in_train' then, conventional scenario")
config = parser.parse_args()
##########################################################
##########################################################
# global logger
log_path = config.log_path + config.dataset + f"({config.look_back}_{config.freq_span}_{config.horizon})" + "_" + f"{config.id_}" + "_" + f"{config.description}"
if not os.path.exists(log_path):
os.makedirs(log_path)
os.environ["log_loc"] = f"{log_path}"
root_dir = os.getcwd()
logging.basicConfig(filename=os.path.join(root_dir, f'{log_path}/log_all.txt'), level=logging.INFO,
format = '%(asctime)s - %(name)s - %(message)s')
logger = logging.getLogger('In main')
logger.info(f"********************* Experiment: Long-term Forecasting *********************")
logger.info(f"Description: {config.description}")
config.model_save_path = os.path.join(log_path,"checkpoints")
config.plots_save_path = os.path.join(log_path,"plots")
config.his_save_path = os.path.join(log_path,"hist")
args = vars(config)
print('------------ Options -------------')
for k, v in sorted(args.items()):
logger.info('%s: %s' % (str(k), str(v)))
print('-------------- End ----------------')
main(config)