batch_size should be a positive integer value, but got batch_size=0

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4

Apache License 2.0

4.47k stars 1.49k forks source link

Open Ronald-Kray opened 3 years ago

Ronald-Kray commented 3 years ago

batch_size should be a positive integer value, but got batch_size=0

How can I fix it?

huafeihuayu commented 3 years ago

batch_size=config.batch // config.subdivisions 修改cfg文件 Cfg.batch = 2 Cfg.subdivisions = 1 Cfg.batch>Cfg.subdivisions