havakv / torchtuples

Training neural networks in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can I use lr_scheduler of Pytorch in model.fit ?

kyuchoi opened this issue · 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

This question is currently discussed in the pycox repo havakv/pycox#103