Skip to content

Instantly share code, notes, and snippets.

@pshashk
pshashk / unet.jl
Last active Jun 10, 2020
Flux unet
View unet.jl
using Flux
downsample(c_in, c_out) = Chain(
Conv((4, 4), c_in => c_out, stride = 2, pad = 1),
BatchNorm(c_out, relu)
)
upsample(c_in, c_out) = Chain(
ConvTranspose((4, 4), c_in => c_out, stride = 2, pad = 1),
BatchNorm(c_out, relu)
You can’t perform that action at this time.