change the name

lasgroup · Oct 3, 2023 · 9c8f70c · 9c8f70c
1 parent ee68d66
commit 9c8f70c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sim_transfer/rl/rl_on_offline_data.py b/sim_transfer/rl/rl_on_offline_data.py
@@ -346,7 +346,7 @@ def prepare_policy_from_offline_data(self,
         directory = os.path.join(wandb.run.dir, 'models')
         if not os.path.exists(directory):
             os.makedirs(directory)
-        model_path = os.path.join('models', 'bnn_svgd_model_on_20_000_points.pkl')
+        model_path = os.path.join('models', 'bnn_model.pkl')
         with open(os.path.join(wandb.run.dir, model_path), 'wb') as handle:
             pickle.dump(bnn_model, handle)
         wandb.save(os.path.join(wandb.run.dir, model_path), wandb.run.dir)