aidevmin
aidevmin
@marcoslucianops Thanks. Do you mean change this line? https://github.com/marcoslucianops/DeepStream-Yolo/blob/9bda315ee0834ca0fb2d7f6b5f34c0a69ddc24e0/config_infer_primary.txt#L10 I will try it and inform to you.
@Michelvl92 Can you run pruning yolov7? Please share information with me. I had error ` Comparison exception: The values for attribute 'shape' do not match: torch.Size([]) != torch.Size([1, 1, 40,...
@VainF Thanks, I think so. We need to keep the last layer related to number of classes and the first layer. I have one more questions. Do you have any...
@VainF Thank you so much for quick response. I got it.
@VainF Do you have any recommed for number of epochs with sparse training? If it is large, so it take much time for normal training + sparse training before pruning.
@MarkusDrange Thank you so much for suggestion. I am going to try it later. Instead of using nni, I used torch-pruning. Could you share experience with pruning yolov7 by using...
@MarkusDrange Can you save pruned yolov7 model by using `torch.save(model, )`?
@MarkusDrange Thanks. I have one more question. Could you finetuning with multiple GPUs after pruning?
Thank @VainF for quick response. It means that I need to run file `yolov7_train_pruned.py`. Is it right? I saw file `yolov7_train_pruned.py` file too, but input of this file is `yolov7_training.pt`,...
Thanks a lot. I will try it.