From 56bf3edf4ed32c1584b4fa921ea050a1863842eb Mon Sep 17 00:00:00 2001
From: Cathy0908 <30484308+Cathy0908@users.noreply.github.com>
Date: Mon, 12 Dec 2022 11:31:07 +0800
Subject: [PATCH] fix export ssl models (#257)

---
 easycv/apis/export.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/easycv/apis/export.py b/easycv/apis/export.py
index dc5ac330..9075c61a 100644
--- a/easycv/apis/export.py
+++ b/easycv/apis/export.py
@@ -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',