ldctbench.methods.cnn10.network

Model(args)

Bases: Module

CNN-10 network. As described in the original paper, we use a simple 3 layer CNN with following parameters

Layer 1: Kernel size 9x9, number of filters 64
Layer 2: Kernel size 3x3, number of filters 32
Layer 3: Kernel size 5x5, number of filters 1.

They use ReLU as nonlinearity and no BatchNorm.