nn_loss: Fastai custom loss

Description Usage Arguments Value

View source: R/experimental.R

Description

Fastai custom loss

Usage

1
nn_loss(loss_fn, name = "Custom_Loss")

Arguments

loss_fn

pass custom model function

name

set name for nn_module

Value

None


fastai documentation built on Oct. 25, 2021, 5:08 p.m.