nn_loss: Fastai custom loss

View source: R/experimental.R

nn_lossR Documentation

Fastai custom loss

Description

Fastai custom loss

Usage

nn_loss(loss_fn, name = "Custom_Loss")

Arguments

loss_fn

pass custom model function

name

set name for nn_module

Value

None


henry090/fastai documentation built on April 14, 2024, 9:28 a.m.