create_cnn_model: Create_cnn_model

View source: R/cnn_learner.R

create_cnn_modelR Documentation

Create_cnn_model

Description

Create custom convnet architecture using 'arch', 'n_in' and 'n_out'

Usage

create_cnn_model(
  arch,
  n_out,
  cut = NULL,
  pretrained = TRUE,
  n_in = 3,
  init = nn()$init$kaiming_normal_,
  custom_head = NULL,
  concat_pool = TRUE,
  lin_ftrs = NULL,
  ps = 0.5,
  bn_final = FALSE,
  lin_first = FALSE,
  y_range = NULL
)

Arguments

arch

a model architecture

n_out

number of outs

cut

cut

pretrained

pretrained model or not

n_in

input shape

init

initializer

custom_head

custom head

concat_pool

concatenate pooling

lin_ftrs

linear fiters

ps

parameter server

bn_final

batch normalization final

lin_first

linear first

y_range

y_range

Value

None


fastai documentation built on March 21, 2022, 9:07 a.m.