Epoch 1/20 Traceback (most recent call last): File "ocs.py", line 172, in model.fit(train_generator(train_tensors, train_labels,batch_size=batch_size), File "