ResBlock: ResBlock

View source: R/blocks.R

ResBlockR Documentation

ResBlock

Description

Resnet block from 'ni' to 'nh' with 'stride'

Usage

ResBlock(
  expansion,
  ni,
  nf,
  stride = 1,
  groups = 1,
  reduction = NULL,
  nh1 = NULL,
  nh2 = NULL,
  dw = FALSE,
  g2 = 1,
  sa = FALSE,
  sym = FALSE,
  norm_type = 1,
  act_cls = nn$ReLU,
  ndim = 2,
  ks = 3,
  pool = AvgPool(),
  pool_first = TRUE,
  padding = NULL,
  bias = NULL,
  bn_1st = TRUE,
  transpose = FALSE,
  init = "auto",
  xtra = NULL,
  bias_std = 0.01,
  dilation = 1,
  padding_mode = "zeros"
)

Arguments

expansion

decoder

ni

number of linear inputs

nf

number of features

stride

stride number

groups

groups number

reduction

reduction

nh1

out channels 1

nh2

out channels 2

dw

dw paramer

g2

g2 block

sa

sa parameter

sym

symmetric

norm_type

normalization type

act_cls

activation

ndim

dimension number

ks

kernel size

pool

pooling type, Average, Max

pool_first

pooling first

padding

padding

bias

bias

bn_1st

batch normalization 1st

transpose

transpose

init

initializer

xtra

xtra

bias_std

bias standard deviation

dilation

dilation number

padding_mode

padding mode

Value

Block object


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