diff --git a/example/ssd_coco2017/eval.py b/example/ssd_coco2017/eval.py index c0af504de2..d5e0d86b67 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)