Merge branch 'main' of github.com:april-tools/squared-npcs

april-tools · Jan 16, 2025 · 6074a44 · 6074a44
2 parents 5047c0a + e60f75b
commit 6074a44
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,10 +1,10 @@
-torch>=2.0.0
+torch~=2.2.0
 pandas<2.0.0
 torchtext
 portalocker
 scikit-learn
 zuko
-numpy
+numpy<2.0.0
 scipy
 torchvision
 seaborn

diff --git a/src/scripts/experiment.py b/src/scripts/experiment.py
@@ -23,7 +23,7 @@
 parser.add_argument('--tboard-path', default='', type=str, help="The Tensorboard path, empty to disable")
 parser.add_argument('--log-distribution', action='store_true', default=False,
                     help="Whether to log the learned distribution")
-parser.add_argument('--log-frequency', default=100,
+parser.add_argument('--log-frequency', default=100, type=int,
                     help="The frequency for logging distributions")
 parser.add_argument('--wandb-path', default='', type=str, help="The W&B path, empty to disable")
 parser.add_argument('--wandb-project', default='born-pcs', type=str, help="The W&B project")