nn_module: Fastai NN module

Description Usage Arguments Value

View source: R/experimental.R

Description

Fastai NN module

Usage

1
nn_module(model_fn, name = "Custom_Model", gpu = TRUE)

Arguments

model_fn

pass custom model function

name

set name for nn_module

gpu

move model to GPU

Value

None


fastai documentation built on July 28, 2021, 5:06 p.m.