| nn_module {fastai} | R 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
[Package fastai version 2.2.2 Index]