Skip to content

Commit 761f790

Browse files
committed
bugfix res101 cpu
1 parent f428de9 commit 761f790

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/depthmap.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ def run_depthmap(processed, outpath, inputimages, inputnames, compute_device, mo
218218
model = RelDepthModel(backbone='resnext101')
219219
model.load_state_dict(strip_prefix_if_present(checkpoint['depth_model'], "module."), strict=True)
220220
del checkpoint
221-
torch.cuda.empty_cache()
221+
devices.torch_gc()
222222

223223
# load merge network if boost enabled
224224
if boost:

0 commit comments

Comments
 (0)