search

Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch
Apache License 2.0
498 stars 128 forks source link

Maybe a bug in assembler/trainval.py #63

Closed felixfuu closed 3 years ago

felixfuu commented 3 years ago

The models in train/val engine are different, and it's maybe a bug in the validation phase.

https://github.com/Media-Smart/vedadet/blob/main/vedadet/assembler/trainval.py#L27 https://github.com/Media-Smart/vedadet/blob/main/vedadet/assembler/trainval.py#L51 @mileistone

mileistone commented 3 years ago

We noticed it and solved this problem by singleton_arg in https://github.com/Media-Smart/vedadet/blob/aaf3b3bc3c7944aba1cc28138165d403023a9152/vedadet/models/builder.py#L44