torchtuples icon indicating copy to clipboard operation
torchtuples copied to clipboard

Can I use lr_scheduler of Pytorch in model.fit ?

Open kyuchoi opened this issue 3 years ago • 1 comments

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

kyuchoi avatar Oct 20 '21 00:10 kyuchoi

This question is currently discussed in the pycox repo https://github.com/havakv/pycox/issues/103

havakv avatar Oct 24 '21 10:10 havakv