From e7e461ff0a59aa71bb01654e7e67183872796c78 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=85=86=E9=98=B3?=
 <SENSETIME\zhangzhaoyang@hk0014000243l.domain.sensetime.com>
Date: Fri, 10 Dec 2021 17:08:54 +0800
Subject: [PATCH] fix evaluate.py load model

---
 evaluate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/evaluate.py b/evaluate.py
index 5c3f502..e095ef9 100644
--- a/evaluate.py
+++ b/evaluate.py
@@ -257,7 +257,7 @@ def main_worker():
     model = net.InpaintGenerator().to(device)
     model_path = args.ckpt
     data = torch.load(args.ckpt, map_location=device)
-    model.load_state_dict(data['netG'])
+    model.load_state_dict(data)
     print('loading from: {}'.format(args.ckpt))
     model.eval()