Fine tune with `freeze` for `freeze_epochs` then with `unfreeze` from `epochs` using discriminative LR
fine_tune(
object,
epochs,
base_lr = 0.002,
freeze_epochs = 1,
lr_mult = 100,
pct_start = 0.3,
div = 5,
...
)
learner/model
epoch number
base learning rate
freeze epochs number
learning rate multiply
start percentage
divide
additional arguments
None