mirrors
/
EasyCV
mirror of https://github.com/alibaba/EasyCV.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix export ssl models (#257)

pull/259/head
Cathy0908 2022-12-12 11:31:07 +08:00 committed by GitHub
parent 5583101471
commit 56bf3edf4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
easycv/apis/export.py
View File

@ -359,6 +359,7 @@ def _export_swav(model, cfg, filename):
model_config = dict(
type='Classification',
pretrained=False, # avoid loading default pretrained backbone model
backbone=tbackbone,
)
@ -426,6 +427,7 @@ def _export_moco(model, cfg, filename):
model_config = dict(
type='Classification',
pretrained=False, # avoid loading default pretrained backbone model
backbone=replace_syncbn(cfg.model.backbone),
head=dict(
type='ClsHead',
@ -481,6 +483,7 @@ def _export_moby(model, cfg, filename):
model_config = dict(
type='Classification',
pretrained=False, # avoid loading default pretrained backbone model
backbone=replace_syncbn(cfg.model.backbone),
head=dict(
type='ClsHead',