def main(*args): train.run_training(training_params())
import argparse import tensorflow as tf from jonasz.progressive_infogan import train import importlib if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('--experiment_module', help='Experiment module to take training_params from.', required=True) parser.add_argument( '--output_dir', help='GCS location to write checkpoints and export models', required=True) parser.add_argument( '--job-dir', help='unused', default='', ) args = parser.parse_args() arguments = args.__dict__ output_dir = arguments.pop('output_dir') experiment_module = importlib.import_module(arguments['experiment_module']) training_params = experiment_module.training_params(is_gcloud=True, output_dir=output_dir) train.run_training(training_params)