create_head: Create_head

Description Usage Arguments Value

View source: R/vision_learner.R

Description

Model head that takes 'nf' features, runs through 'lin_ftrs', and out 'n_out' classes.

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
create_head(
  nf,
  n_out,
  lin_ftrs = NULL,
  ps = 0.5,
  concat_pool = TRUE,
  bn_final = FALSE,
  lin_first = FALSE,
  y_range = NULL
)

Arguments

nf

number of features

n_out

number of out features

lin_ftrs

linear features

ps

parameter server

concat_pool

concatenate pooling

bn_final

batch normalization final

lin_first

linear first

y_range

y_range

Value

None


fastai documentation built on July 28, 2021, 5:06 p.m.