Ejemplo n.º 1
0
def test_autoencode_vgg16_celeba():

    args = config.config([
        '--config', '../configs/autoencoder/vgg16/celeba.yaml', '--epochs',
        '5', '--dataroot', '../data', '--dataset_test_len', '256',
        '--dataset_train_len', '256'
    ])
    best_loss = train_autoencoder.main(args)
    assert best_loss < 0.04
Ejemplo n.º 2
0
def test_autoencode_fc_mnist():

    args = config.config([
        '--config', '../configs/autoencoder/fc/mnist.yaml', '--epochs', '80',
        '--dataroot', '../data', '--dataset_test_len', '256',
        '--dataset_train_len', '256'
    ])
    best_loss = train_autoencoder.main(args)
    assert best_loss < 1.3
Ejemplo n.º 3
0
def test_autoencode_resnet_cifar10():

    args = config.config([
        '--config', '../configs/autoencoder/resnet/cifar10.yaml', '--epochs',
        '20', '--optim_lr', '1e-5', '--dataroot', '../data',
        '--dataset_test_len', '256', '--dataset_train_len', '256', '--seed',
        '0'
    ])
    best_loss = train_autoencoder.main(args)
    assert best_loss < 1.2