remove trailing benchmark

recursionpharma · Aug 16, 2023 · d56f962 · d56f962
1 parent 49c8d41
commit d56f962
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/gflownet/algo/trajectory_balance.py b/src/gflownet/algo/trajectory_balance.py
@@ -459,8 +459,6 @@ def compute_batch_losses(
             "logZ": log_Z.mean(),
             "loss": loss.item(),
         }
-        if self.cfg.do_subtb and self.cfg.cum_subtb:
-            info["subtb_diff"] = subtb_diff
         return loss, info
 
     def _init_subtb(self, dev):