From ac2afc4cb9c21a08e237c396e146f0fd5c6b28c8 Mon Sep 17 00:00:00 2001 From: David Rotermund Date: Sun, 17 Nov 2024 02:55:56 +0100 Subject: [PATCH] Update stage_4_process.py --- stage_4_process.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stage_4_process.py b/stage_4_process.py index f046ea9..b8e4382 100644 --- a/stage_4_process.py +++ b/stage_4_process.py @@ -498,7 +498,7 @@ def process_trial( volume_heartbeat_cpu, _, _ = torch.linalg.svd( heartbeat_ts.cpu(), full_matrices=False ) - volume_heartbeat = volume_heartbeat_cpu.to(heartbeat_ts.data, copy=True) + volume_heartbeat = volume_heartbeat_cpu.to(heartbeat_ts.device, copy=True) del volume_heartbeat_cpu volume_heartbeat = volume_heartbeat[:, 0]