delete print statements

src/sparcscore/ml/datasets.py

@@ -298,8 +298,6 @@ def __getitem__(self, idx):
             idx = idx.tolist() # convert tensor to list
 
         data_item = self.data_locator[idx] # get the data info for the current index, such as target, handle id, and row
-
-        #print(f"Getting data for index {idx}...")
 
         if self.select_channel is not None: # select a specific channel
             cell_tensor = self.handle_list[data_item[1]][data_item[2], self.select_channel] 

src/sparcscore/ml/models.py

@@ -182,24 +182,10 @@ def vgg(cfg, in_channels,  **kwargs):
         return model
 
     def forward(self, x):
-        print("x.shape before norm", x.shape)
         x = self.norm(x)
-        print("x.shape after norm", x.shape)
         x = self.features(x)
-        print("x.shape after features", x.shape)
-
         x = torch.flatten(x, 1)
-        print("x.shape after flatten", x.shape)
-
         x = self.classifier(x)
-        print("x.shape after classifier", x.shape)
-
-        print(x, x.shape)
-
-        x_sq = torch.squeeze(x)
-
-        print(x_sq, x_sq.shape)
-
         return x
 
 ### CAE Model Architecture

src/sparcscore/ml/plmodels.py

@@ -204,7 +204,6 @@ def configure_optimizers(self):
 
     def training_step(self, batch):
         data, target = batch
-        print("target training ", target)
         target = target.unsqueeze(1) # Add dimension for regression
         output = self.network(data) # Forward pass, only one output
         loss = F.mse_loss(output, target) # L2 loss