OlafenwaMoses/ImageAI

TypeError

CodeWithChai opened this issue · 0 comments

from imageai.Classification.Custom import ClassificationModelTrainer
model_trainer = ClassificationModelTrainer()
model_trainer.setModelTypeAsResNet50()
model_trainer.setDataDirectory(r"/content/drive/MyDrive/pets")
model_trainer.trainModel(num_objects=10, num_experiments=100, enhance_data=True,batch_size=32, show_network_summary=True)

--
I keep on getting this error despite trying everything, any solutions ?
TypeError: ClassificationModelTrainer.trainModel() got an unexpected keyword argument 'num_objects'