parent
c6cf781fa6
commit
637958a910
|
@ -123,6 +123,9 @@ class BaseTask(metaclass=ABCMeta):
|
|||
load_checkpoint(model, model_checkpoint, map_location=self.device)
|
||||
|
||||
model = revert_sync_batchnorm(model)
|
||||
if hasattr(model, 'backbone') and hasattr(model.backbone,
|
||||
'switch_to_deploy'):
|
||||
model.backbone.switch_to_deploy()
|
||||
model = model.to(self.device)
|
||||
model.eval()
|
||||
return model
|
||||
|
|
|
@ -215,7 +215,7 @@ models:
|
|||
- name: MobileOne
|
||||
metafile: configs/mobileone/metafile.yml
|
||||
model_configs:
|
||||
- configs/mobileone/deploy/mobileone-s0_deploy_8xb128_in1k.py
|
||||
- configs/mobileone/mobileone-s0_8xb32_in1k.py
|
||||
pipelines:
|
||||
- *pipeline_trt_static_fp16
|
||||
- *pipeline_ort_dynamic_fp32
|
||||
|
|
Loading…
Reference in New Issue