Liu ke
Results
3
comments of
Liu ke
good，solved my problem thx!
Same problem with 6g memories.
> File "test.py", line 67, in test > mask = ndimage.interpolation.zoom(mask, scale, order=1) Shold be replaced by `mask = ndimage.interpolation.zoom(mask.cpu(), scale, order=1)` cause the mask is in GPU by default.