swintransformer训练提示RuntimeError: expected scalar type Float but found Half

positive666 / yolo_research

based on yolo-high-level project (detect\pose\classify\segment\):include yolov5\yolov7\yolov8\ core ,improvement research ,SwintransformV2 and Attention Series. training skills, business customization, engineering deployment C

GNU General Public License v3.0

758 stars 146 forks source link

swintransformer训练提示RuntimeError: expected scalar type Float but found Half #19

Closed hellozzk closed 2 years ago

wwwwwbh commented 2 years ago

如果想不报这个错的话，可以把train.py中val.run()的 half 参数的 not 去掉，就用float32了大约在380行和440行

positive666 commented 2 years ago

No description provided.

SWIN 要在参数上写参数的，因为不支持FP16，

positive666 commented 2 years ago

代码大更新了修复了一些BUG和缺少位置变量参数优化更新的问题