add forgotten keyword in train_line.py

GFNOrg · Apr 2, 2024 · 89c72b5 · 89c72b5
1 parent bff764d
commit 89c72b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tutorials/examples/train_line.py b/tutorials/examples/train_line.py
@@ -229,6 +229,7 @@ def train(
             env,
             n_samples=batch_size,
             save_estimator_outputs=True,
+            save_logprobs=False,
             scale_factor=scale_schedule[iteration],  # Off policy kwargs.
         )
         training_samples = gflownet.to_training_samples(trajectories)