Attribute error while training an efficientnet model

Asked Jun 21 '21 at 05:28

Active Jun 21 '21 at 07:32

Viewed 732 times

I'm trying to train a prediction model using EfficientnetB4 and coming across this error:

AttributeError                            Traceback (most recent call last)
<ipython-input-16-5066d78634f9> in <module>()
----> 1 X_train, X_val, X_test, y_train, y_val, y_test = get_features(model, train_it, validate_it)

<ipython-input-10-437f64953231> in get_features(base_model, train, validate)
      1 def get_features(base_model, train, validate):
      2     X_train = base_model.predict(train)
----> 3     y_train = train.classes
      4 
      5     X_val = base_model.predict(validate)

AttributeError: 'BatchDataset' object has no attribute 'classes'

The BatchDataset seems to be an object within efficientnet.

edited Jun 21 '21 at 07:32

Trong Van

asked Jun 21 '21 at 05:28

Stephen Myers

What did you mean by ``train.classes``? As the error indicates it has no ``classes`` attribute. – Kaveh Jun 21 '21 at 06:55
Could you share the minimal reproducible code. Thanks! – Jul 02 '21 at 02:47

Attribute error while training an efficientnet model

0 Answers0