Skip to content
Snippets Groups Projects
Commit bc050e59 authored by anzhengqi's avatar anzhengqi
Browse files

modify network squeezenet parameter error to fix bug

parent 1d8fe960
No related branches found
No related tags found
No related merge requests found
......@@ -102,7 +102,7 @@ YOLOv3使用DarkNet53执行特征提取,这是YOLOv2中的Darknet-19和残差
wget https://pjreddie.com/media/files/darknet53.conv.74
```
- 可以运行model_zoo/utils/hccl_tools/路径下的hccl_tools.py脚本生成hccl_8p.json文件,下面指令中参数"[0, 8)"表示生成0-7的8卡hccl_8p.json文件。
- 可以运行models/utils/hccl_tools/路径下的hccl_tools.py脚本生成hccl_8p.json文件,下面指令中参数"[0, 8)"表示生成0-7的8卡hccl_8p.json文件。
- 该命令生成的json文件名为 hccl_8p_01234567_{host_ip}.json, 为了表述方便,统一使用hccl_8p.json表示该json文件。
```command
......
......@@ -47,7 +47,7 @@ if args.device_target == "Ascend":
if __name__ == '__main__':
net = squeezenet(num_classes=num_classes)
param_dict = load_checkpoint(args.ckpt_file)
param_dict = load_checkpoint(args.checkpoint_path)
load_param_into_net(net, param_dict)
input_data = Tensor(np.zeros([args.batch_size, 3, args.height, args.width], np.float32))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment