create_mlp {fastai}R Documentation

Create_mlp

Description

A simple model builder to create a bunch of BatchNorm1d, Dropout and Linear layers, with “'act_fn“' activations.

Usage

create_mlp(ni, nout, linear_sizes = c(500, 500, 500))

Arguments

ni

number of input channels

nout

output shape

linear_sizes

linear output sizes

Value

model


[Package fastai version 2.2.2 Index]