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
I have read the README carefully. 我已经仔细阅读了README上的操作指引。
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等配置文件训练自定义数据集。)
I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码,重新运行之后,问题仍不能解决。
Search before asking
I have searched the YOLOv6 issues and found no similar questions.
Question
我用pytorch的fx图模式对模型进行量化报错
preds, s_featmaps = self.qat_model(images)
File "/opt/conda/lib/python3.7/site-packages/torch/fx/graph_module.py", line 616, in wrapped_call
raise e.with_traceback(None)
RuntimeError: expected scalar type Float but found Half
我没看到哪里有对模型做.half()也没看到哪对数据做.half(),并且打印出来的images.dtype和qat_model.dtype都是float32,好奇怪啊
Additional
No response
The text was updated successfully, but these errors were encountered:
Before Asking
I have read the README carefully. 我已经仔细阅读了README上的操作指引。
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等配置文件训练自定义数据集。)
I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码,重新运行之后,问题仍不能解决。
Search before asking
Question
我用pytorch的fx图模式对模型进行量化报错
preds, s_featmaps = self.qat_model(images)
File "/opt/conda/lib/python3.7/site-packages/torch/fx/graph_module.py", line 616, in wrapped_call
raise e.with_traceback(None)
RuntimeError: expected scalar type Float but found Half
我没看到哪里有对模型做.half()也没看到哪对数据做.half(),并且打印出来的images.dtype和qat_model.dtype都是float32,好奇怪啊
Additional
No response
The text was updated successfully, but these errors were encountered: