Skip to content

TypeError: train() got an unexpected keyword argument 'callbacks' #403

@fferlito

Description

@fferlito

I used to use this repo using callbacks, so I could terminate the training with early stopping. I recently discovered that this is not an option anymore, and the parameter is not working anymore. Is there a reason of this?

This is the code that I used to run:

model = mobilenet_unet(n_classes=11,  input_height=1024, input_width=1024, channels=3  )
# When using custom callbacks, the default checkpoint saver is removed
callbacks = [
    ModelCheckpoint(
                filepath=checkpoint_path,
                save_weights_only=True,
                verbose=True,
                save_best_only=True,
                monitor='val_accuracy'
            ),
    EarlyStopping(patience=50,
                  monitor='val_accuracy')
]

print("Start training...")
model.train(
    train_images =  "train/backup/train_images/",
    train_annotations = "train/backup/train_labels/",
    epochs=100000,
    batch_size=5,
    gen_use_multiprocessing=False,
    do_augment=True,
    augmentation_name='aug_all',
    validate=True,
    val_images="train/test_images/",
    val_annotations="train//test_labels/",
    val_batch_size=1,
    steps_per_epoch=14,
    val_steps_per_epoch=19,
    #steps_per_epoch=733,
    #val_steps_per_epoch=73,
    callbacks=callbacks
)

From which I get:
model.train( TypeError: train() got an unexpected keyword argument 'callbacks'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions