Update utils/actor_utils.py

utils/actor_utils.py

@@ -54,7 +54,7 @@ def select_data(actor, index, include_embeddings=True, include_projections=True)
 
     def select_attributes(obj):
         for name, value in obj.__dict__.items():
-            if isinstance(value, torch.Tensor) and value.size(0) >= index.max().item():
+            if isinstance(value, torch.Tensor) and value.dim() > 0 and value.size(0) >= index.max().item():
                 setattr(obj, name, value[index])
 
     select_attributes(actor.fleet)