create_mlp: Create_mlp

Description Usage Arguments Value

View source: R/time_series_model.R

Description

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

Usage

1
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


fastai documentation built on Oct. 25, 2021, 5:08 p.m.