unet_config: Unet_config

View source: R/unet.R

unet_configR Documentation

Unet_config

Description

Convenience function to easily create a config for 'DynamicUnet'

Usage

unet_config(
  blur = FALSE,
  blur_final = TRUE,
  self_attention = FALSE,
  y_range = NULL,
  last_cross = TRUE,
  bottle = FALSE,
  act_cls = nn()$ReLU,
  init = nn()$init$kaiming_normal_,
  norm_type = NULL
)

Arguments

blur

blur is used to avoid checkerboard artifacts at each layer.

blur_final

blur final is specific to the last layer.

self_attention

self_attention determines if we use a self attention layer at the third block before the end.

y_range

If y_range is passed, the last activations go through a sigmoid rescaled to that range.

last_cross

last cros

bottle

bottle

act_cls

activation

init

initializer

norm_type

normalization type

Value

None


fastai documentation built on March 31, 2023, 11:41 p.m.