grappa_qdmr_aug.jsonnet

{
    name: "grappa-aug",
    logdir: "logdir/%s" % self.name,
    model_config: "text2qdmr/configs/text2qdmr-base.jsonnet",
    model_config_args: {
        save_name: 'grappa_qdmr_train',
        save_data_path: "text2qdmr/preproc_data/",
        spider_data_path: "data/spider/",
        break_data_path: "data/break/",
        grounding_path: "data/break/groundings/",
        grounding_mode: "first",
        train_max_values_from_database: 25,
        eval_max_values_from_database: 25,
        augment_at_iter_shuffle_tables: true,
        augment_at_iter_shuffle_columns: true,
        augment_at_iter_shuffle_values: true,
        augment_at_iter_shuffle_sort_dir: true,
        augment_at_iter_shuffle_compsup_op: true,
        bs: 6,
        num_batch_accumulated: 4,
        pretrained_version: "grappa",
        max_steps: 81000,
        lr: 7.44e-4,
        bert_lr: 3e-6,
        att: 1,
        end_lr: 0,
        update_name: 'relational_transformer',
        sc_link: true,
        merge_sc_link: false,
        cv_link: false,
        use_relations: true,
        use_graph_relations: true,
        use_type_relations: true,
        use_online_data_processing: true,
        num_dataloading_workers: 3, # half of the batch size
    },

    eval_name: "%s_beam_%d" % [self.name, self.eval_beam_size],
    eval_output: "__LOGDIR__/ie_dirs",
    eval_beam_size: 1,
    eval_steps: [81000],
    eval_section: ["val", "test"],
    vis_dir: "full_val",
    eval_tb_dir: "runs_viz/%s_%s" % [self.name, self.eval_section],
    eval_strict_decoding: true,
    # virtuoso_server: 'http://link_to_virtuoso_server/'
}