vagan_train.py

# Authors:
# Christian F. Baumgartner (c.f.baumgartner@gmail.com)


import logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s')

from vagan.model_vagan import vagan


def main():

    # Select experiment below:

    # from vagan.experiments import adni_experiment as exp_config
    from vagan.experiments import synthetic_experiment as exp_config

    # Get Data
    if exp_config.data_identifier == 'synthetic':
        from data.synthetic_data import synthetic_data as data_loader
    elif exp_config.data_identifier == 'adni':
        from data.adni_data import adni_data as data_loader
    else:
        raise ValueError('Unknown data identifier: %s' % exp_config.data_identifier)

    data = data_loader(exp_config)

    # Build VAGAN model
    vagan_model = vagan(exp_config=exp_config, data=data, fixed_batch_size=exp_config.batch_size)

    # Train VAGAN model
    vagan_model.train()


if __name__ == '__main__':

    main()