shinya7y / UniverseNet

USB: Universal-Scale Object Detection Benchmark (BMVC 2022)
Apache License 2.0
422 stars 54 forks source link

Multi gpu training #24

Closed sh7jacobi closed 3 years ago

sh7jacobi commented 3 years ago

Is it possible to train with more than one gpu. When I try running bash tools/dist_train.sh configs/gflv2/gflv2_visdrone.py 2 I get AttributeError: 'ConfigDict' object has no attribute 'total_epochs'

sh7jacobi commented 3 years ago

I have found my mistake)