75 lines
3.2 KiB
Python
Executable File
75 lines
3.2 KiB
Python
Executable File
def print_args(args):
|
|
print("\033[1m" + "Basic Config" + "\033[0m")
|
|
print(
|
|
f" {'Task Name:':<20}{args.task_name:<20}{'Is Training:':<20}{args.is_training:<20}"
|
|
)
|
|
print(f" {'Model ID:':<20}{args.model_id:<20}{'Model:':<20}{args.model:<20}")
|
|
print()
|
|
|
|
print("\033[1m" + "Data Loader" + "\033[0m")
|
|
print(f" {'Data:':<20}{args.data:<20}{'Root Path:':<20}{args.root_path:<20}")
|
|
print(
|
|
f" {'Data Path:':<20}{args.data_path:<20}{'Features:':<20}{args.features:<20}"
|
|
)
|
|
print(f" {'Target:':<20}{args.target:<20}{'Freq:':<20}{args.freq:<20}")
|
|
print(f" {'Checkpoints:':<20}{args.checkpoints:<20}")
|
|
print()
|
|
|
|
if args.task_name in ["long_term_forecast", "short_term_forecast"]:
|
|
print("\033[1m" + "Forecasting Task" + "\033[0m")
|
|
print(
|
|
f" {'Seq Len:':<20}{args.seq_len:<20}{'Label Len:':<20}{args.label_len:<20}"
|
|
)
|
|
print(
|
|
f" {'Pred Len:':<20}{args.pred_len:<20}{'Seasonal Patterns:':<20}{args.seasonal_patterns:<20}"
|
|
)
|
|
print(f" {'Inverse:':<20}{args.inverse:<20}")
|
|
print()
|
|
|
|
if args.task_name == "imputation":
|
|
print("\033[1m" + "Imputation Task" + "\033[0m")
|
|
print(f" {'Mask Rate:':<20}{args.mask_rate:<20}")
|
|
print()
|
|
|
|
if args.task_name == "anomaly_detection":
|
|
print("\033[1m" + "Anomaly Detection Task" + "\033[0m")
|
|
print(f" {'Anomaly Ratio:':<20}{args.anomaly_ratio:<20}")
|
|
print()
|
|
|
|
print("\033[1m" + "Model Parameters" + "\033[0m")
|
|
print(f" {'Top k:':<20}{args.top_k:<20}{'Num Kernels:':<20}{args.num_kernels:<20}")
|
|
print(f" {'Enc In:':<20}{args.enc_in:<20}{'Dec In:':<20}{args.dec_in:<20}")
|
|
print(f" {'C Out:':<20}{args.c_out:<20}{'d model:':<20}{args.d_model:<20}")
|
|
print(f" {'n heads:':<20}{args.n_heads:<20}{'e layers:':<20}{args.e_layers:<20}")
|
|
print(f" {'d layers:':<20}{args.d_layers:<20}{'d FF:':<20}{args.d_ff:<20}")
|
|
print(f" {'Moving Avg:':<20}{args.moving_avg:<20}{'Factor:':<20}{args.factor:<20}")
|
|
print(f" {'Distil:':<20}{args.distil:<20}{'Dropout:':<20}{args.dropout:<20}")
|
|
print(f" {'Embed:':<20}{args.embed:<20}{'Activation:':<20}{args.activation:<20}")
|
|
print()
|
|
|
|
print("\033[1m" + "Run Parameters" + "\033[0m")
|
|
print(f" {'Num Workers:':<20}{args.num_workers:<20}{'Itr:':<20}{args.itr:<20}")
|
|
print(
|
|
f" {'Train Epochs:':<20}{args.train_epochs:<20}{'Batch Size:':<20}{args.batch_size:<20}"
|
|
)
|
|
print(
|
|
f" {'Patience:':<20}{args.patience:<20}{'Learning Rate:':<20}{args.learning_rate:<20}"
|
|
)
|
|
print(f" {'Des:':<20}{args.des:<20}{'Loss:':<20}{args.loss:<20}")
|
|
print(f" {'Lradj:':<20}{args.lradj:<20}{'Use Amp:':<20}{args.use_amp:<20}")
|
|
print()
|
|
|
|
print("\033[1m" + "GPU" + "\033[0m")
|
|
print(f" {'Use GPU:':<20}{args.use_gpu:<20}{'GPU:':<20}{args.gpu:<20}")
|
|
print(
|
|
f" {'Use Multi GPU:':<20}{args.use_multi_gpu:<20}{'Devices:':<20}{args.devices:<20}"
|
|
)
|
|
print()
|
|
|
|
print("\033[1m" + "De-stationary Projector Params" + "\033[0m")
|
|
p_hidden_dims_str = ", ".join(map(str, args.p_hidden_dims))
|
|
print(
|
|
f" {'P Hidden Dims:':<20}{p_hidden_dims_str:<20}{'P Hidden Layers:':<20}{args.p_hidden_layers:<20}"
|
|
)
|
|
print()
|