zhangming8 comments

Results 30 comments of


                                            zhangming8

Is there any mistake in network architecture ?

Hi, do you solve the problems ? I meet the same question

训练时 cuda 指定问题

哦，设置gpu的时候是用逗号,隔开的吗？gpus="3,4,5"，或者修改opt.gpus = "3,4,5"有效吗

训练时 cuda 指定问题

> hi： > 支持一波，写的很棒，已经用上了，感谢。发现一个小问题，训练时改变 train.sh gpus 参数并不能指定 gpu。这个bug现在已经修复，并且支持多gpu训练

bug: fixed path of coco 128

You still using my dataset path. Maybe, you not modify 'dataset_path' to "/mnt/d/coco_obj" in "train.sh"

mapo0.5 or map0.5:0.95

ap_0_5指的是amp iou = 0.5的时候。ap指的是map0.5:0.95

for batch_i, batch in enumerate(train_loader): inps, targets, img_info, ind = batch

我试了一下，没问题，你用的最新的代码吗

进行evaluate时候出现问题

拉一下最新代码，这个后处理之前用的cpu，但是预测结果是在gpu上，现在这个已经修复了。https://github.com/zhangming8/yolox-pytorch/blob/main/models/post_process.py#L22-L24

> 新的代码在训练设定300个epoch，在284个epoch的时候又出现了**AttributeError: 'MosaicDetection' object has no attribute 'close_mosaic'**这个问题不好意思，修bug的时候这个没注意到，需要把https://github.com/zhangming8/yolox-pytorch/blob/main/train.py#L137 里面的.dataset去掉。中途断掉可以参考train.sh里面resume接着训练