Flux.BatchNorm doesn't accept Int32 as argument

General Usage
2

n_layers seems a misleading argument name, it should be n_channels. Beside that, can’t you simply use an Int. Flux.BatchNorm(3). Or convert it: Flux.BatchNorm(Int(n_layers))

1 Like