davrot/gevi
1
0
Fork 0
mirror of https://github.com/davrot/gevi.git synced 2025-04-20 22:26:43 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update stage_4_process.py

Browse source
This commit is contained in:
David Rotermund 2024-11-17 02:55:56 +01:00 • committed by GitHub
parent e53c8e9fdf
commit ac2afc4cb9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
stage_4_process.py
View file

@ -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]