!1290 adapt to modification of initializer

Merge pull request !1290 from chengxb7532/master
pull/1290/MERGE
mindspore-ci-bot 6 years ago committed by Gitee
commit a3b9c238cc

@ -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