Fix utilization decl order

thatguy11325 · Jun 24, 2024 · a1bd73d · a1bd73d
1 parent ae9f0a7
commit a1bd73d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
@@ -142,6 +142,7 @@ class CleanPuffeRL:
     def __post_init__(self):
         seed_everything(self.config.seed, self.config.torch_deterministic)
         if self.config.verbose:
+            self.utilization = Utilization()
             print_dashboard(
                 self.config.env,
                 self.utilization,
@@ -154,8 +155,6 @@ def __post_init__(self):
                 clear=True,
             )
 
-            self.utilization = Utilization()
-
         self.vecenv.async_reset(self.config.seed)
         obs_shape = self.vecenv.single_observation_space.shape
         obs_dtype = self.vecenv.single_observation_space.dtype