Python main Examples

Example #1

File: test_autoencoder.py Project: ZombaSY/iic-1

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

Example #2

File: test_autoencoder.py Project: ZombaSY/iic-1

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

Example #3

File: test_autoencoder.py Project: ZombaSY/iic-1

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