Browse Source

!1290 adapt to modification of initializer

Merge pull request !1290 from chengxb7532/master
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
a3b9c238cc
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)
print("Load Checkpoint!")
param_dict = load_checkpoint(ckpt_path)
net.init_parameters_data()
load_param_into_net(net, param_dict)

net.set_train(False)


Loading…
Cancel
Save