Mohamed Martini

> I just changed the following in the train.py: > > ```python > model = tf.keras.Model(input_tensor, output_tensors) #Already existing line > model.load_weights('./yolov3') #This is the new line > ``` >...

Try specifying the number of classes, e.g.: `python load_weights.py --num_classes 4`