I meet some bug

[ToChangeBetter]

your work is excellent.I want to reproduce this paper. When I run the (history = model.fit(generator = training_generator, epochs=EPOCHS, validation_data=validation_generator,steps_per_epoch = (len(partition['train']) // train_params['batch_size']) ,use_multiprocessing=True,callbacks =all_callbacks( path + '.hdf5' ),workers=16)).it appear (ValueError: The output of the "schedule" function should be float.).Can you help me ?