From 162075e4960530c621ca90d747009bfc61b6b45a Mon Sep 17 00:00:00 2001 From: Federico Pizarro Bejarano Date: Mon, 18 Nov 2024 09:42:36 -0500 Subject: [PATCH] Minor update --- experiments/mpsc/mpsc_experiment.py | 2 +- experiments/mpsc/train_all_models.sh | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/experiments/mpsc/mpsc_experiment.py b/experiments/mpsc/mpsc_experiment.py index 46bd96dfc..56f6e3180 100644 --- 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 index 5ef76e8fb..dc7152cab 100755 --- 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