Removed float32

GreeleyGroup · Sep 23, 2023 · a9d3d7a · a9d3d7a
1 parent 57196b1
commit a9d3d7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models.py b/src/models.py
@@ -129,7 +129,7 @@ def forward(self, data_objects):
         # For each data object
         for i, data in enumerate(data_objects):
             # Apply initial transform
-            conv_data = self.init_transform[i](data.x.to(torch.float32)) 
+            conv_data = self.init_transform[i](data.x) 
 
             # Apply convolutional layers
             for layer in self.conv_layers[i]: