Fixed MeasureSamplesPerSecondCallback and changed configs (#138)

* Update wandb.yaml

* Update wandb_cluster.yaml

* Fix condition to check tensor dimensions

* Set `entity` to `null` in config instead of "MyOrganisation"

---------

Co-authored-by: Fabrice Normandin <[email protected]>

Author: 3rdCore

project/algorithms/callbacks/samples_per_second.py

@@ -190,7 +190,7 @@ def get_num_samples(self, batch: BatchType) -> int:
             return next(
                 v.shape[0]
                 for v in optree.tree_leaves(batch)  # type: ignore
-                if isinstance(v, torch.Tensor) and v.ndim > 1
+                if isinstance(v, torch.Tensor) and v.ndim >= 1
             )
         raise NotImplementedError(
             f"Don't know how many 'samples' there are in batch of type {type(batch)}"

project/configs/trainer/logger/wandb.yaml

@@ -2,6 +2,7 @@
 
 wandb:
   _target_: lightning.pytorch.loggers.wandb.WandbLogger
+  entity: null  # Optional. It can be useful to set this explicitly if you use a different organization account for this project.
   project: "ResearchTemplate"
   name: ${name}
   save_dir: "."

project/configs/trainer/logger/wandb_cluster.yaml

@@ -1,6 +1,7 @@
 defaults:
   - wandb
 wandb:
+  entity: null  # Optional. It can be useful to set this explicitly if you use a different organization account for this project.
   project: "ResearchTemplate"
   # TODO: Use the Orion trial name?
   name: ${oc.env:SLURM_JOB_ID}_${oc.env:SLURM_PROCID}