Skip to contents

FashionMNIST CNN architecture resolved to concrete shapes (hmfmn)

Source: R/geron_w4a_native.R
morie_geron_fashion_mnist.Rd

Output sizes delegated to morie_geron_conv_output_size.

Usage

morie_geron_fashion_mnist(
  epochs = 10,
  lr = 0.001,
  batch_size = 32,
  n_classes = 10,
  input_size = 28,
  filters = c(32, 64)
)

Arguments

epochs, lr, batch_size, n_classes, input_size

Config.

filters

Channels per conv block.

Value

List with layers, total_params, flatten_dim, fc_share, class_names.

Examples

morie_geron_fashion_mnist()
#> $layers
#> $layers[[1]]
#> $layers[[1]]$kind
#> [1] "conv"
#> 
#> $layers[[1]]$kernel
#> [1] 3
#> 
#> $layers[[1]]$channels
#> [1] 32
#> 
#> $layers[[1]]$out
#> [1] 26
#> 
#> $layers[[1]]$params
#> [1] 320
#> 
#> 
#> $layers[[2]]
#> $layers[[2]]$kind
#> [1] "pool"
#> 
#> $layers[[2]]$kernel
#> [1] 2
#> 
#> $layers[[2]]$channels
#> [1] 32
#> 
#> $layers[[2]]$out
#> [1] 13
#> 
#> $layers[[2]]$params
#> [1] 0
#> 
#> 
#> $layers[[3]]
#> $layers[[3]]$kind
#> [1] "conv"
#> 
#> $layers[[3]]$kernel
#> [1] 3
#> 
#> $layers[[3]]$channels
#> [1] 64
#> 
#> $layers[[3]]$out
#> [1] 11
#> 
#> $layers[[3]]$params
#> [1] 18496
#> 
#> 
#> $layers[[4]]
#> $layers[[4]]$kind
#> [1] "pool"
#> 
#> $layers[[4]]$kernel
#> [1] 2
#> 
#> $layers[[4]]$channels
#> [1] 64
#> 
#> $layers[[4]]$out
#> [1] 5
#> 
#> $layers[[4]]$params
#> [1] 0
#> 
#> 
#> $layers[[5]]
#> $layers[[5]]$kind
#> [1] "flatten"
#> 
#> $layers[[5]]$out
#> [1] 1600
#> 
#> $layers[[5]]$channels
#> [1] 64
#> 
#> $layers[[5]]$params
#> [1] 0
#> 
#> 
#> $layers[[6]]
#> $layers[[6]]$kind
#> [1] "fc"
#> 
#> $layers[[6]]$out
#> [1] 128
#> 
#> $layers[[6]]$channels
#> [1] 128
#> 
#> $layers[[6]]$params
#> [1] 204928
#> 
#> 
#> $layers[[7]]
#> $layers[[7]]$kind
#> [1] "fc"
#> 
#> $layers[[7]]$out
#> [1] 10
#> 
#> $layers[[7]]$channels
#> [1] 10
#> 
#> $layers[[7]]$params
#> [1] 1290
#> 
#> $layers[[7]]$activation
#> [1] "softmax"
#> 
#> 
#> 
#> $total_params
#> [1] 225034
#> 
#> $conv_params
#> [1] 18816
#> 
#> $fc_params
#> [1] 206218
#> 
#> $fc_share
#> [1] 0.916386
#> 
#> $flatten_dim
#> [1] 1600
#> 
#> $class_names
#>  [1] "T-shirt/top" "Trouser"     "Pullover"    "Dress"       "Coat"       
#>  [6] "Sandal"      "Shirt"       "Sneaker"     "Bag"         "Ankle boot" 
#> 
#> $output_shape
#> [1] 10
#> 
#> $training_config
#> $training_config$epochs
#> [1] 10
#> 
#> $training_config$lr
#> [1] 0.001
#> 
#> $training_config$batch_size
#> [1] 32
#> 
#> 
#> $steps_per_epoch
#> [1] 1875
#> 
#> $estimate
#> [1] 225034
#> 
#> $n
#> [1] 7
#> 
#> $method
#> [1] "FashionMNIST CNN resolved to concrete shapes; output sizes delegated to grcos"
#>