YOLOX icon indicating copy to clipboard operation
YOLOX copied to clipboard

There are no logs outputs information like loss etc. during the training Process

Open jayphone17 opened this issue 2 years ago • 1 comments

I use

python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 8 -b 128 --fp16 -o -c weights/yolox_s.pth

for multi GPUS training but there are no logs outputting

` 2022-07-06 14:44:58 | INFO | yolox.core.trainer:203 - ---> start train epoch1

2022-07-06 14:45:07 | INFO | yolox.core.trainer:356 - Save weights to ./YOLOX_outputs/yolox_voc_s

2022-07-06 14:45:07 | INFO | yolox.core.trainer:203 - ---> start train epoch2

2022-07-06 14:45:14 | INFO | yolox.core.trainer:356 - Save weights to ./YOLOX_outputs/yolox_voc_s

`

while is all fine using only 1 GPU

python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 0 -b 4 --fp16 -o -c weights/yolox_s.pth

like:

` 2022-07-06 15:07:47 | INFO | yolox.core.trainer:203 - ---> start train epoch1

2022-07-06 15:07:50 | INFO | yolox.core.trainer:253 - epoch: 1/300, iter: 10/128, mem: 10091Mb, iter_time: 0.328s, data_time: 0.001s, total_loss: 15.5, iou_loss: 3.6, l1_loss: 0.0, conf_loss: 9.6, cls_loss: 2.3, lr: 3.815e-06, size: 640, ETA: 3:29:45

2022-07-06 15:07:53 | INFO | yolox.core.trainer:253 - epoch: 1/300, iter: 20/128, mem: 10091Mb, iter_time: 0.311s, data_time: 0.002s, total_loss: 15.7, iou_loss: 3.5, l1_loss: 0.0, conf_loss: 10.1, cls_loss: 2.1, lr: 1.526e-05, size: 768, ETA: 3:24:10 `

jayphone17 avatar Jul 06 '22 07:07 jayphone17

have u solved the problem?

wendalegood avatar Jul 14 '22 01:07 wendalegood