unable to train

[lchunleo]

Hi

thanks for sharing. i have perform a training using the code (with the directory change to reflect the actual position of the dataset) but i encountered..

Please use Model.fit, which supports generators.
WARNING:tensorflow:Model failed to serialize as JSON. Ignoring... Layer YoloLayer has arguments in __init__ and therefore must override get_config.
Epoch 1/100
....

Error encountered after execution:

ValueError: tf.function-decorated function tried to create variables on non-first call.

i have raised the same query on imageai, if it is answered there, will close the thread here. thank you