Fastai NN module
nn_module(model_fn, name = "Custom_Model", gpu = TRUE)
model_fn
pass custom model function
name
set name for nn_module
gpu
move model to GPU
None