create_cnn_model: Create_cnn_model

In fastai: Interface to 'fastai'

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 June 22, 2024, 11:15 a.m.