Update modularStarEncoder.py

modularStarEncoder.py

@@ -209,13 +209,12 @@ def get_pooling_mask(
     if DEVICE<0:
         DEVICE = "cpu"
     ranges = torch.arange(input_ids.size(1)).repeat(input_ids.size(0), 1)
-    ranges.to(DEVICE)
-    repeated_idx.to(DEVICE)
-    print(ranges,repeated_idx)
-    print(ranges.get_device(),repeated_idx.get_device())
+    ranges = ranges.to(DEVICE)
+
+
 
     pooling_mask = (repeated_idx <= ranges).long()
-    print(pooling_mask.get_device())
+
 
     return pooling_mask