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)
def train_omniglot(): config = get_Omniglot_config() data_generator = DataGenerator(config) maml = MAML(config) maml.train_manager(data_generator)