nn_module: Fastai NN module

View source: R/experimental.R

nn_moduleR Documentation

Fastai NN module

Description

Fastai NN module

Usage

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 June 22, 2024, 11:15 a.m.