Fastai custom loss

nn_loss(loss_fn, name = "Custom_Loss")

Arguments

loss_fn

pass custom model function

name

set name for nn_module

Value

None