antsnormflows.nets.cnn

Classes

ConvNet2d(channels, kernel_size[, leaky, ...])

2D Convolutional Neural Network with leaky ReLU nonlinearities

ConvNet3d(channels, kernel_size[, leaky, ...])

3D Convolutional Neural Network with leaky ReLU nonlinearities