Create_mlp

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

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