diff --git a/experiments/run_m4.py b/experiments/run_m4.py index ca07745f..fc127e29 100644 --- a/experiments/run_m4.py +++ b/experiments/run_m4.py @@ -20,7 +20,7 @@ def parse_args(): parser = ArgumentParser(description="Run time series forecasting model in BasicTS framework!") - parser.add_argument("-c", "--config", default="baselines/MLP/M4_base.py", help="training config template") + parser.add_argument("-c", "--config", default="baselines/MLP/M4.py", help="training config template") parser.add_argument("-g", "--gpus", default="0", help="visible gpus") parser.add_argument("--save_evaluation", default=True, help="if save evaluation results") parser.add_argument("--save_prediction", default=False, help="if save prediction results")