Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

执行稀疏训练时候出现了cfg文件不匹配的情况 #3

Open
AllenZYJ opened this issue Feb 25, 2022 · 0 comments
Open

执行稀疏训练时候出现了cfg文件不匹配的情况 #3

AllenZYJ opened this issue Feb 25, 2022 · 0 comments

Comments

@AllenZYJ
Copy link

作者你好,我在执行稀疏训练的时候,发现cfg文件的某些weight读出来是个空的sequential(),是cfg和pt不匹配的缘故吗:

command:

python train_sparsity.py --img 640 --batch 16 --epochs 10 --data yolo1104.yaml --cfg models/yolov5l.yaml --weights yolov5l.pt --name base_0225_sparsity --scale 0.001 --prune 1

log:

Model Summary: 468 layers, 46165219 parameters, 46165219 gradients, 108.0 GFLOPs

Transferred 606/613 items from yolov5l.pt
Scaled weight_decay = 0.0005
optimizer: SGD with parameter groups 101 weight, 104 weight (no decay), 104 bias
WARNING: smart bias initialization failure.
WARNING: smart bias initialization failure.
WARNING: smart bias initialization failure.
Traceback (most recent call last):
File "train_sparsity.py", line 672, in
main(opt)
File "train_sparsity.py", line 569, in main
train(opt.hyp, opt, device, callbacks)
File "train_sparsity.py", line 210, in train
copy_weight_v6(model, cfg_model)
File "/backup/yolov5_prune_0119/Normal_Dete/yolov5-v6/utils/model_transfer.py", line 85, in copy_weight_v6
idx=copy_c3(cspnet2,model.module_list,idx, round(3 * depth_multiple))
File "/backup/yolov5_prune_0119/Normal_Dete/yolov5-v6/utils/model_transfer.py", line 40, in copy_c3
idx=copy_conv_idx(c3_src.m[i].cv1, module_lists[idx],idx)
File "/backup/yolov5_prune_0119/Normal_Dete/yolov5-v6/utils/model_transfer.py", line 22, in copy_conv_idx
copy_conv(conv_src,conv_dst)
File "/backup/yolov5_prune_0119/Normal_Dete/yolov5-v6/utils/model_transfer.py", line 12, in copy_conv
conv_dst[0] = conv_src.conv
File "/root/miniconda3/envs/yolo/lib/python3.7/site-packages/torch/nn/modules/container.py", line 108, in setitem
key: str = self._get_item_by_idx(self._modules.keys(), idx)
File "/root/miniconda3/envs/yolo/lib/python3.7/site-packages/torch/nn/modules/container.py", line 96, in _get_item_by_idx
raise IndexError('index {} is out of range'.format(idx))
IndexError: index 0 is out of range

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant