create_mlp: Create_mlp

View source: R/time_series_model.R

create_mlpR 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


fastai documentation built on June 22, 2024, 11:15 a.m.