From ef83ad73a839a555fb6655385238c961657c32b6 Mon Sep 17 00:00:00 2001 From: Joseph Viviano Date: Wed, 17 Jul 2024 17:49:54 -0400 Subject: [PATCH] note left for self --- tutorials/examples/train_hypergrid_multinode.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tutorials/examples/train_hypergrid_multinode.py b/tutorials/examples/train_hypergrid_multinode.py index b6eac423..98e1e4ff 100644 --- a/tutorials/examples/train_hypergrid_multinode.py +++ b/tutorials/examples/train_hypergrid_multinode.py @@ -373,6 +373,7 @@ def validate_hypergrid( ) # Add the mode counting metric. + # TODO: This is not the same as what is done in `train_hypergrid`. states, scale = visited_terminating_states.tensor, env.scale_factor normalized_states = ((states * scale) - (scale / 2) * (env.height - 1)).abs()