ScaledYOLOv4 NaN output from the first layer when training with GPU

NaN output from the first layer when training with GPU

Open hoangphucITJP opened this issue 3 years ago • 1 comments

As the title. When I train on the CPU, it works fine. But training on GPU produces nan after the first block.

Please advise. Thanks.

Jun 09 '21 11:06 hoangphucITJP

@hoangphucITJP, any leads on this one?

May 17 '22 15:05 kXborg