AttributeError: 'function' object has no attribute 'set_model'

Question

I am fitting a model using keras and passed the callbacks list to the model fit_generator but encounterd the following error.Please help.

AttributeError: 'function' object has no attribute 'set_model'

Code snippet :

    from keras.callbacks import LearningRateScheduler
    import numpy as np
    from keras import optimizers
    from keras.callbacks import *

    def lr_schedule(epoch):
        lrate = 0.1
        if epoch > 50:
            lrate = 0.01
        elif epoch > 75:
            lrate = 0.001       
        return lrate


    filepath="latest_weight_ckpt_{epoch:02d}_{val_acc:.2f}.hdf5"
    model_ckpt = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, 
    save_best_only=True, mode='max')
    #lists = [model_ckpt]


    #model_checkpoint= ModelCheckpoint("weights/DenseNet-BC-100-12- 
    CIFAR100.h5", monitor="val_acc", 
    save_best_only=True,save_weights_only=True)

    callbacks_list=[lr_schedule, model_ckpt]

    model.fit_generator(train_generator, epochs=25, 
    steps_per_epoch=200,verbose=1, validation_steps=200, 
    validation_data=validation_generator,callbacks=callbacks_list)

Anna Krogager · Accepted Answer · 2019-03-31 08:14:31Z

6

The problem is that you are passing a python function (lr_schedule) as a callback instead of a Keras callback object. You should instead use the Keras LearningRateScheduler callback as follows:

lr = LearningRateScheduler(lr_schedule)
callbacks_list=[lr, model_ckpt]

answered Mar 31, 2019 at 8:14

Anna Krogager

3,59819 silver badges24 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

AttributeError: 'function' object has no attribute 'set_model'

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related