def resnet_tiny_cpu(): hparams = resnet.resnet_base() hparams.layer_sizes = [2, 2, 2, 2] hparams.use_nchw = False return hparams
def resnet_tiny_cpu(): hparams = resnet.resnet_base() hparams.layer_sizes = [2, 2, 2, 2] hparams.num_filters = [10, 20, 30, 40] hparams.use_nchw = False return hparams