Fastai NN module

Fastai NN module

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