Closed Adeel-Intizar closed 4 years ago
Try to add yolo.batch_size = 8 before yolo.make_model() It seems to be out of memory.
yolo.batch_size = 8
yolo.make_model()