Skip to content

Fix transfer of data from GPU to CPU after computation#239

Open
marcovarrone wants to merge 1 commit into
MathOnco:mainfrom
marcovarrone:fix-208-gpu-compatibility
Open

Fix transfer of data from GPU to CPU after computation#239
marcovarrone wants to merge 1 commit into
MathOnco:mainfrom
marcovarrone:fix-208-gpu-compatibility

Add `.cpu()` calls before converting PyTorch tensors to NumPy arrays.

bd97743
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs