修复配置错误

.vscode/settings.json

@@ -4,8 +4,8 @@
     "python-envs.pythonProjects": [
         {
             "path": "data/SolarEnergy",
-            "envManager": "ms-python.python:system",
-            "packageManager": "ms-python.python:pip"
+            "envManager": "ms-python.python:conda",
+            "packageManager": "ms-python.python:conda"
         }
     ]
 }

config/REPST/METR-LA.yaml

@@ -41,7 +41,7 @@ train:
   batch_size: 16
   early_stop: true
   early_stop_patience: 15
-  epochs: 1
+  epochs: 100
   grad_norm: false
   loss_func: mae
   lr_decay: true
@@ -52,7 +52,7 @@ train:
   real_value: true
   weight_decay: 0
   debug: false
-  output_dim: 100
+  output_dim: 1
   log_step: 1000
   plot: false
   mae_thresh: None