From 6fd3f5bbab6b9dcfe0c1d0b7a73c216a312066a6 Mon Sep 17 00:00:00 2001 From: saleml Date: Tue, 2 Apr 2024 14:06:32 +0400 Subject: [PATCH] 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, )