From d1d1961bb3a416b1133ca1dcb3b3c1207a12ce93 Mon Sep 17 00:00:00 2001 From: Joseph Viviano Date: Sun, 31 Mar 2024 12:36:56 -0400 Subject: [PATCH 1/3] added matplotlib --- pyproject.toml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 36947af0..77074fdc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -44,6 +44,7 @@ tqdm = { version = "*", optional = true } wandb = { version = "*", optional = true } scikit-learn = {version = "*", optional = true } scipy = { version = "*", optional = true } +matplotlib = { version = "*", optional = true } [tool.poetry.extras] dev = [ @@ -61,11 +62,12 @@ dev = [ "flake8", ] -scripts = ["tqdm", "wandb", "scikit-learn", "scipy"] +scripts = ["tqdm", "wandb", "scikit-learn", "scipy", "matplotlib"] all = [ "black", "flake8", + "matplotlib", "myst-parser", "pre-commit", "pytest", From 34d8232044216becdf532f9f432413a10fe5156d Mon Sep 17 00:00:00 2001 From: Joseph Viviano Date: Sun, 31 Mar 2024 12:41:55 -0400 Subject: [PATCH 2/3] train_line trains closer to convergence by default --- tutorials/examples/train_line.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/examples/train_line.py b/tutorials/examples/train_line.py index ccbbf1cf..3621e02e 100644 --- a/tutorials/examples/train_line.py +++ b/tutorials/examples/train_line.py @@ -297,7 +297,7 @@ def train( gflownet, environment, lr_base=1e-3, - n_trajectories=1.28e6, + n_trajectories=5e6, batch_size=256, exploration_var_starting_val=exploration_var_starting_val, ) From 6fd3f5bbab6b9dcfe0c1d0b7a73c216a312066a6 Mon Sep 17 00:00:00 2001 From: saleml Date: Tue, 2 Apr 2024 14:06:32 +0400 Subject: [PATCH 3/3] revert back to 1.28e6 --- tutorials/examples/train_line.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/examples/train_line.py b/tutorials/examples/train_line.py index 3621e02e..ccbbf1cf 100644 --- a/tutorials/examples/train_line.py +++ b/tutorials/examples/train_line.py @@ -297,7 +297,7 @@ def train( gflownet, environment, lr_base=1e-3, - n_trajectories=5e6, + n_trajectories=1.28e6, batch_size=256, exploration_var_starting_val=exploration_var_starting_val, )