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()
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()