train过程中出现Longtensor转换问题的错误解决办法

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。

GNU General Public License v3.0

861 stars 150 forks source link

Open luohao136486939 opened 1 year ago

luohao136486939 commented 1 year ago

yolov7-pytorch/nets/yolo_training.py中第507行，将gain变量转换为 LongTenso，具体如下，gain = gain.long()

bubbliiiing commented 1 year ago

你是1.12？

luohao136486939 commented 1 year ago

你是1.12？

阿导，你的1.12指的这个代码么？我就是用你master的这个版本