YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Apache License 2.0
9.27k
stars
2.18k
forks
source link
TypeError: expected Tensor as element 1 in argument 0, but got int #1059
我训练yolox_nano模型,其中yolox_voc_nano.py文件用于设置训练参数,self.max_epoch用于设置最大训练轮数。当我给这个变量赋值为5或10时,它可以正常训练;当赋值为50甚至更大的值时,会报如上的错误。想寻求解决办法。
Command: python tools/train.py -f exps/example/yolox_voc/yolox_voc_nano.py -c yolox_nano.pth --batch-size 8 --devices 1 --fp16
Error:![MD6NM@R%VH3TDCP9Q0_`(RP](https://user-images.githubusercontent.com/71818703/148674288-08a2ac04-9fe7-4b25-826a-5dea0915697d.png)
yolox_voc_nano.py: