Skip to content

GitLab

"models/vscode:/vscode.git/clone" did not exist on "d04051e3e2af0865967a9cc2c28b48f94af9945e"
Unverified Commit 84855b37 authored by Tai-Wang's avatar Tai-Wang Committed by GitHub
Browse files

Fix the memory-leak problem in init_detector (#1045)

parent 5ff7c6ac
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/apis/inference.py
View file @ 84855b37
...@@ -57,7 +57,7 @@ def init_model(config, checkpoint=None, device='cuda:0'): ...@@ -57,7 +57,7 @@ def init_model(config, checkpoint=None, device='cuda:0'):
config.model.train_cfg = None config.model.train_cfg = None
model = build_model(config.model, test_cfg=config.get('test_cfg')) model = build_model(config.model, test_cfg=config.get('test_cfg'))
if checkpoint is not None: if checkpoint is not None:
checkpoint = load_checkpoint(model, checkpoint) checkpoint = load_checkpoint(model, checkpoint, map_location='cpu')
if 'CLASSES' in checkpoint['meta']: if 'CLASSES' in checkpoint['meta']:
model.CLASSES = checkpoint['meta']['CLASSES'] model.CLASSES = checkpoint['meta']['CLASSES']
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment