Minor update

Federico-PizarroBejarano · Nov 18, 2024 · 162075e · 162075e
1 parent 0132626
commit 162075e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/experiments/mpsc/mpsc_experiment.py b/experiments/mpsc/mpsc_experiment.py
@@ -72,7 +72,7 @@ def run(plot=True, training=False, n_episodes=1, n_steps=None, curr_path='.', in
 
     if config.algo in ['ppo', 'sac', 'safe_explorer_ppo', 'cpo']:
         # Load state_dict from trained.
-        ctrl.load(f'{curr_path}/models/rl_models/{system}/{task}/{config.algo}/{model}/model_latest.pt')
+        ctrl.load(f'{curr_path}/models/rl_models/{model}/model_latest.pt')
 
         # Remove temporary files and directories
         shutil.rmtree(f'{curr_path}/temp', ignore_errors=True)

diff --git a/experiments/mpsc/train_all_models.sh b/experiments/mpsc/train_all_models.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
-./train_model.sbatch True 1 1
+sbatch train_model.sbatch True 1 1
 for MPSC_COST_HORIZON in 2 5 10 20; do
     for DECAY_FACTOR in 0.25 0.5 0.75 1; do
-        ./train_model.sbatch True  $MPSC_COST_HORIZON $DECAY_FACTOR
-        # ./train_model.sbatch False $MPSC_COST_HORIZON $DECAY_FACTOR
+        sbatch train_model.sbatch True  $MPSC_COST_HORIZON $DECAY_FACTOR
+        sbatch train_model.sbatch False $MPSC_COST_HORIZON $DECAY_FACTOR
     done
 done