wutong16/DistributionBalancedLoss

bug: args.gpus is missing

VamosC opened this issue 3 years ago · 0 comments

VamosC commented 3 years ago

args.gpus is missing

DistributionBalancedLoss/tools/train.py

Line 59 in a3ecaa9

cfg.gpus = args.gpus

maybe add parser.add_argument('--gpus', default=1) in the parse_args()