You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
作者你好,我在执行稀疏训练的时候,发现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
The text was updated successfully, but these errors were encountered: