MrComNguoi

I also have this problem and I don't know why, I have 500 images as training images but the len(train_loader) is only 8. Why is that?

You have to go to the 'CenterNet/src/lib/detectors/base_detector.py' file and replace `if opt.gpus[0] >= 0: opt.device = torch.device('cuda') else: opt.device = torch.device('cpu')` with `opt.device = torch.device('cpu')` only. Also delete (or uncomment)...