adapt to modification of initializer

2020-05-20 16:04:29 +08:00 · 2020-05-20 16:04:29 +08:00 · 1c86cf496a
parent 20478b574b
commit 1c86cf496a
1 changed files with 1 additions and 0 deletions
--- 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)