Skip to content

Can I use lr_scheduler of Pytorch in model.fit ? #24

@kyuchoi

Description

@kyuchoi

First of all, thank you for your great works !!

Can I use lr_scheduler of Pytorch in model.fit ?
I've noticed that there is lr_scheduler.py in this repo, however, I couldn't find how to use it in examples.
If it's possible, then how can I modify the following code to use the scheduler for learning rate?

log = model.fit(x_train, y_train, batch_size, epochs, callbacks, val_data=val)

Many thanks

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