DistributionBalancedLoss icon indicating copy to clipboard operation
DistributionBalancedLoss copied to clipboard
verified publisher icon wutong16

bug: args.gpus is missing

Open VamosC opened this issue 4 years ago • 0 comments

args.gpus is missing https://github.com/wutong16/DistributionBalancedLoss/blob/a3ecaa9021a920fcce9fdafbd7d83b51bf526af8/tools/train.py#L59

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

VamosC avatar Sep 09 '21 13:09 VamosC