mlcommons / training

Reference implementations of MLPerf™ training benchmarks
https://mlcommons.org/en/groups/training
Apache License 2.0
1.57k stars 548 forks source link

[MaskRCNN bug] make_data_loader() method should only return data_loaders[0] when training #670

Open Xiao-Yamin opened 11 months ago

Xiao-Yamin commented 11 months ago

Question: The following error is reported when training MaskRCNN: image

Solution: In training\object_detection\pytorch\maskrcnn_benchmark\data\build.py, make_data_loader() method should only return data_loaders[0] when training, but now it returns data_loaders[0] and iterations_per_epoch to report this error. image