slimgroup / InvertibleNetworks.jl

A Julia framework for invertible neural networks
MIT License
149 stars 22 forks source link

residual block w/o default subsampling #5

Closed philippwitte closed 4 years ago

philippwitte commented 4 years ago

Default residual w/o spatial subsampling.

Default for residual block and HINT layers/networks: Layer 1: 3 x 3, padding 1, stride 1 Layer 2: 3 x 3, padding 1, stride 1 Layer 3: 3 x 3, padding 1, stride 1

Default for glow layer and network: Layer 1: 3 x 3, padding 1, stride 1 Layer 2: 1 x 1, padding 0, stride 1 Layer 3: 3 x 3, padding 1, stride 1

Default for iRIM layer and network: Layer 1: 4 x 4, padding 0, stride 4 (down) Layer 2: 3 x 3, padding 1, stride 1 Layer 3: 4 x 4, padding 0, stride 4 (up)