diff --git a/experiments/long_horizon_benchmarks/run_eval.py b/experiments/long_horizon_benchmarks/run_eval.py index bf25952..b715257 100644 --- a/experiments/long_horizon_benchmarks/run_eval.py +++ b/experiments/long_horizon_benchmarks/run_eval.py @@ -31,6 +31,7 @@ FLAGS = flags.FLAGS _BATCH_SIZE = flags.DEFINE_integer("batch_size", 64, "Batch size for the randomly sampled batch") _DATASET = flags.DEFINE_string("dataset", "etth1", "The name of the dataset.") + _MODEL_PATH = flags.DEFINE_string("model_path", "./timesfm_q10_20240501", "The name of the dataset.") _DATETIME_COL = flags.DEFINE_string("datetime_col", "date",