add results dir (#1044)

Author: rayg1234

src/fairchem/core/_cli_hydra.py

@@ -52,6 +52,7 @@
 
 LOG_DIR_NAME = "logs"
 CHECKPOINT_DIR_NAME = "checkpoints"
+RESULTS_DIR = "results"
 CONFIG_FILE_NAME = "canonical_config.yaml"
 PREEMPTION_STATE_DIR_NAME = "preemption_state"
 
@@ -90,6 +91,7 @@ class Metadata:
     commit: str
     log_dir: str
     checkpoint_dir: str
+    results_dir: str
     config_path: str
     preemption_checkpoint_dir: str
     cluster_name: str
@@ -120,6 +122,7 @@ def __post_init__(self) -> None:
             checkpoint_dir=os.path.join(
                 self.run_dir, self.timestamp_id, CHECKPOINT_DIR_NAME
             ),
+            results_dir=os.path.join(self.run_dir, self.timestamp_id, RESULTS_DIR),
             config_path=os.path.join(self.run_dir, self.timestamp_id, CONFIG_FILE_NAME),
             preemption_checkpoint_dir=os.path.join(
                 self.run_dir,

src/fairchem/core/common/utils.py

@@ -1458,11 +1458,11 @@ def get_timestamp_uid() -> str:
 @torch.no_grad()
 def tensor_stats(name: str, x: torch.Tensor) -> dict:
     return {
-        f"{name}.max": x.max(),
-        f"{name}.min": x.min(),
-        f"{name}.std": x.std(),
-        f"{name}.mean": x.mean(),
-        f"{name}.norm": torch.norm(x, p=2),
+        f"{name}.max": x.max().item(),
+        f"{name}.min": x.min().item(),
+        f"{name}.std": x.std().item(),
+        f"{name}.mean": x.mean().item(),
+        f"{name}.norm": torch.norm(x, p=2).item(),
         f"{name}.nonzero_fraction": torch.nonzero(x).shape[0] / float(x.numel()),
     }