From 89c72b5add431fcd8a787323cdc14aee7ee1ffe8 Mon Sep 17 00:00:00 2001 From: saleml Date: Tue, 2 Apr 2024 14:03:27 +0400 Subject: [PATCH] add forgotten keyword in train_line.py --- tutorials/examples/train_line.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tutorials/examples/train_line.py b/tutorials/examples/train_line.py index 0fb9db09..6ce7fde6 100644 --- 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)