예제 #1
0
def test_eval_squad_bidaf():
    config = NestedNamespace()
    config.data_file_path = SYNTHETIC_DATA_PATH
    config.checkpoint_path = "./logs/test/bidaf/checkpoint/model_1.pkl"
    config.cude_devices = None
    set_gpu_env(config)

    experiment = Experiment(Mode.EVAL, config)
    experiment()
예제 #2
0
def test_eval_nlu_bert_for_tok_cls():
    config = NestedNamespace()
    config.data_file_path = SYNTHETIC_DATA_PATH
    config.checkpoint_path = "./logs/test/tok_cls/bert/checkpoint/model_1.pkl"
    config.cude_devices = None
    set_gpu_env(config)

    experiment = Experiment(Mode.EVAL, config)
    experiment()