paddle-paddle - 在 paddledetection 中没有输出的第一个 epoch 后训练停止

Question

这是图片

我在 CPU 上使用相同的模型 Paddledetection 和相同的参数，但在 GPU 上工作正常。它在第一个纪元后停止，没有错误和警告，也没有输出。

C:\Users\Jing\PD>python tools/train.py -c "C:\Users\Jing\PD\configs\ppyolo\ppyolo_mbv3_small_coco.yml" 将注解加载到内存中... 完成 (t=0.01s) 创建索引...索引已创建！W0609 17:55:52.611917 2056 device_context.cc:404] 请注意：设备：0，GPU 计算能力：8.6，驱动程序 API 版本：11.2，运行时 API 版本：11.2 W0609 17:55:52.630895 2056 device_context.cc:422]设备：0，cuDNN 版本：8.1。[06/09 17:55:54] ppdet.utils.download 信息：从https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV3_small_x1_0_ssld_pretrained.pdparams下载 MobileNetV3_small_x1_0_ssld_pretrained.pdparams 100%|███████████████████████████████████████████████ ██████████████████████████████████████████████████ ██████████████████████████████████████████████████ █████████████████████████████████████████████████| 3443/3443 [00:04<00:00, 834.87KB/s] [06/09 17:55:58] ppdet.utils.checkpoint INFO: Finish loading model weights: C:\Users\Jing/.cache/paddle /weights\MobileNetV3_small_x1_0_ssld_pretrained.pdparams [06/09 17:56:00] ppdet.engine INFO: Epoch: [0] [ 0/10] learning_rate: 0.000000 loss_xy: 2.762484 loss_wh: 11.338955 loss_iou: 10.873783 loss_obj: 3044.532715 loss_cls: 22.257437 loss : 3091.765381 eta: 1:00:05 batch_cost: 1.3354 data_cost: 0.0000 ips: 23.9623 images/s

标签： paddle-paddle