update functions for working resnet optim

a-rouxel · Mar 9, 2024 · 660c7ad · 660c7ad
1 parent 71cf609
commit 660c7ad
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/simca/CassiSystem_lightning.py b/simca/CassiSystem_lightning.py
@@ -161,7 +161,8 @@ def generate_filtering_cube(self):
         numpy.ndarray: filtering cube generated according to the optical system & the pattern configuration (R x C x W)
 
         """
-        self.filtering_cube = interpolate_data_on_grid_positions_torch(data=self.pattern.unsqueeze(-1).repeat(1, 1, 1, self.wavelengths.shape[0]),
+
+        self.filtering_cube = interpolate_data_on_grid_positions_torch(data=self.pattern.unsqueeze(-1).repeat(1, 1, 1, self.wavelengths.shape[0]).to(self.device),
                                                                 X_init=self.X_coordinates_propagated_coded_aperture,
                                                                 Y_init=self.Y_coordinates_propagated_coded_aperture,
                                                                 X_target=self.X_detector_coordinates_grid,

diff --git a/simca/functions_acquisition_torch.py b/simca/functions_acquisition_torch.py
@@ -84,6 +84,11 @@ def interpolate_data_on_grid_positions_torch(data, X_init, Y_init, X_target, Y_t
 
     #print(data.shape)
 
+
+    # Ensure tensors are on the correct device and data type
+    device = data.device
+    dtype = torch.float64  # Using double precision for interpolation calculations
+
     # Ensure tensors are on the correct device and data type
     device = data.device
     dtype = torch.float64  # Using double precision for interpolation calculations