mixmo.networks

Networks used in the main paper

Functions

get_network(config_network, config_args)

Return a new instance of network

Modules

mixmo.networks.resnet	Resnet for cifar dataset.
mixmo.networks.wrn	WideResNet network definition ported from https://github.com/meliketoy/wide-resnet.pytorch/blob/master/networks/wide_resnet.py