From 9c7e36de02317e0b0527ff553f91784d8c875d95 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A4=95=E9=99=8C?= <jiangnana.jnn@alibaba-inc.com>
Date: Fri, 9 Dec 2022 15:50:12 +0800
Subject: [PATCH] fix export ssl models

---
 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',