From d1f8dbb30893d406b19ecc1e45f63c037bd5ec72 Mon Sep 17 00:00:00 2001
From: "zuchen.wang" <zuchen.wang@vipshop.com>
Date: Wed, 24 Nov 2021 16:01:04 +0800
Subject: [PATCH] do not save eval_only to config.yaml

---
 projects/Shoe/train_net.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/projects/Shoe/train_net.py b/projects/Shoe/train_net.py
index fec84b1..c6f7bf7 100644
--- a/projects/Shoe/train_net.py
+++ b/projects/Shoe/train_net.py
@@ -24,14 +24,15 @@ def setup(args):
     cfg = get_cfg()
     cfg.merge_from_file(args.config_file)
     cfg.merge_from_list(args.opts)
-    setattr(cfg, 'eval_only', args.eval_only)
-    cfg.freeze()
     default_setup(cfg, args)
     return cfg
 
 
 def main(args):
     cfg = setup(args)
+    cfg.defrost()
+    setattr(cfg, 'eval_only', args.eval_only)
+    cfg.freeze()
 
     if args.eval_only:
         cfg.defrost()