style_learner
(dls
, model
, get_feats
, style_fns
, loss_func
=None
, cbs
=None
, metrics
=None
, opt_func
='Adam'
, lr
=0.001
, splitter
='trainable_params'
, path
=None
, model_dir
='models'
, wd
=None
, wd_bn_bias
=False
, train_bn
=True
, moms
=(0.95, 0.85, 0.95)
)