Weights in ptq training config

oforomar commented 1 year ago

Before Asking

[X] I have read the README carefully. 我已经仔细阅读了README上的操作指引。
[X] I want to train my custom dataset, and I have read the tutorials for training your custom data carefully and organize my dataset correctly; (FYI: We recommand you to apply the config files of xx_finetune.py.) 我想训练自定义数据集，我已经仔细阅读了训练自定义数据的教程，以及按照正确的目录结构存放数据集。（FYI: 我们推荐使用xx_finetune.py等配置文件训练自定义数据集。）
[X] I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

Search before asking

[X] I have searched the YOLOv6 issues and found no similar questions.

Question

Hello, I am trying to do PTQ with yolov6n on a custom dataset. I have followed the reptopt tutorial, and obtained the scales and weights. However I am not sure about the weights that I am supposed to use in the config file in the calib field here:- qat = dict( calib_pt = './assets/v6s_n_calib_max.pt', sensitive_layers_skip = False, sensitive_layers_list=[], )

Additional

No response

xingyueye commented 1 year ago

@oforomar You could generate a v6s_n_calib_max.pt follow the step2 in sensitivity_analyse.py

model_ptq= do_ptq(model, train_loader, args.batch_number, device)
torch.save({'model': model_ptq}, args.weights.replace('.pt', '_calib.pt'))

oforomar commented 1 year ago

Thank you, I will run it and report back.

meituan / YOLOv6