From 1c86cf496a7c0effea1b41cf33f35af7a635ad86 Mon Sep 17 00:00:00 2001 From: chengxianbin Date: Wed, 20 May 2020 16:04:29 +0800 Subject: [PATCH] adapt to modification of initializer --- example/ssd_coco2017/eval.py | 1 + 1 file changed, 1 insertion(+) diff --git a/example/ssd_coco2017/eval.py b/example/ssd_coco2017/eval.py index c0af504de22..d5e0d86b67a 100644 --- a/example/ssd_coco2017/eval.py +++ b/example/ssd_coco2017/eval.py @@ -31,6 +31,7 @@ def ssd_eval(dataset_path, ckpt_path): net = SSD300(ssd_mobilenet_v2(), ConfigSSD(), is_training=False) print("Load Checkpoint!") param_dict = load_checkpoint(ckpt_path) + net.init_parameters_data() load_param_into_net(net, param_dict) net.set_train(False)