Browse Source

adapt to modification of initializer

tags/v0.3.0-alpha
chengxianbin 5 years ago
parent
commit
1c86cf496a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      example/ssd_coco2017/eval.py

+ 1
- 0
example/ssd_coco2017/eval.py View File

@@ -31,6 +31,7 @@ def ssd_eval(dataset_path, ckpt_path):
net = SSD300(ssd_mobilenet_v2(), ConfigSSD(), is_training=False) net = SSD300(ssd_mobilenet_v2(), ConfigSSD(), is_training=False)
print("Load Checkpoint!") print("Load Checkpoint!")
param_dict = load_checkpoint(ckpt_path) param_dict = load_checkpoint(ckpt_path)
net.init_parameters_data()
load_param_into_net(net, param_dict) load_param_into_net(net, param_dict)


net.set_train(False) net.set_train(False)


Loading…
Cancel
Save