Python MAML.train_manager Examples

Example #1

File: main.py Project: aronsar/ganabi

def train_ganabi():
    config_file = './config/ganabi.config.gin'
    gin.parse_config_file(config_file)
    config_obj = TrainConfig()
    # config = config_obj.get_config()

    data_generator = DataGenerator(config_obj)
    maml = MAML(config_obj)
    maml.save_gin_config(config_file)
    maml.train_manager(data_generator)

Example #2

File: main.py Project: aronsar/ganabi

def train_omniglot():
    config = get_Omniglot_config()
    data_generator = DataGenerator(config)
    maml = MAML(config)
    maml.train_manager(data_generator)