Skip to content
Snippets Groups Projects
Commit d21649c9 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!592 modify network squeezenet parameter error to fix bug

Merge pull request !592 from anzhengqi/modify-network-squeezenet-parameter
parents 05251b04 bc050e59
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