From c15745fe811300cb8a7b0b2e7f2672ac83110e3d Mon Sep 17 00:00:00 2001 From: misko Date: Sun, 30 Jun 2024 02:32:44 +0000 Subject: [PATCH] fix tests by moving to float32 --- spf/dataset/spf_dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/spf/dataset/spf_dataset.py b/spf/dataset/spf_dataset.py index 627004be..334a68aa 100644 --- a/spf/dataset/spf_dataset.py +++ b/spf/dataset/spf_dataset.py @@ -449,9 +449,9 @@ def render_session(self, receiver_idx, session_idx): pd = torch_get_phase_diff(data["signal_matrix"]).to(torch.float32) data["x"] = torch.vstack([abs_signal[0], abs_signal[1], pd])[None] - data["y_rad"] = data["ground_truth_theta"][None] - data["y_phi"] = data["ground_truth_phi"][None] - data["craft_y_rad"] = data["craft_ground_truth_theta"][None] + data["y_rad"] = data["ground_truth_theta"][None].to(torch.float32) + data["y_phi"] = data["ground_truth_phi"][None].to(torch.float32) + data["craft_y_rad"] = data["craft_ground_truth_theta"][None].to(torch.float32) # data["y_discrete"] = v5_thetas_to_targets(data["y_rad"], self.nthetas)