Fix path for wandb models

xinpw8 · Jan 15, 2024 · e800153 · e800153
1 parent afd77e4
commit e800153
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/config.yaml b/config.yaml
@@ -607,6 +607,7 @@ ocean:
     num_envs: 8
     batch_rows: 32
     bptt_horizon: 4
+    device: cpu
   env:
     name: squared
 bandit:

diff --git a/demo.py b/demo.py
@@ -219,12 +219,12 @@ def train(args, env_module, make_env):
         args.exp_name = init_wandb(args, env_module).id
     elif args.baseline:
         args.track = True
-        args.exp_name = args.config
+        args.exp_name = f'puf-{pufferlib.__version__}-{args.config}'
         args.wandb_group = f'puf-{pufferlib.__version__}-baseline'
         shutil.rmtree(f'experiments/{args.exp_name}', ignore_errors=True)
         run = init_wandb(args, env_module, name=args.exp_name, resume=False)
         if args.mode == 'evaluate':
-            model_name = f'puf{pufferlib.__version__}-{args.config}_model:latest'
+            model_name = f'puf-{pufferlib.__version__}-{args.config}_model:latest'
             artifact = run.use_artifact(model_name)
             data_dir = artifact.download()
             model_file = max(os.listdir(data_dir))

diff --git a/pufferlib/version.py b/pufferlib/version.py
@@ -1 +1 @@
-__version__ = '0.6.0'
+__version__ = '0.6.1'