'k': None } } if __name__ == '__main__': visible_device_list = '0' visible_devices = '1' memory_fraction = 0.5 model_name = 'attention_document_relation' extra_parser_args_adder = lambda parser: parser extra_args_formatter = lambda args: '' max_ent_len_retrieve = lambda train_doc_store, dev_doc_store, args: 1 max_rel_len_retrieve = lambda train_doc_store, dev_doc_store, args: max( train_doc_store.max_tokens, dev_doc_store.max_tokens) rel_retrieve_params = {'replacement': (ENT_1, ENT_2), 'truncate': False} ent_retrieve_params = {} tensor_dict_map = { 'rel_seq': 'rel_seq', 'seq_len': 'seq_len', 'rel_len': 'rel_len', 'target_rel': 'target_rel' } run_model(visible_device_list, visible_devices, memory_fraction, model_name, extra_parser_args_adder, extra_args_formatter, max_ent_len_retrieve, max_rel_len_retrieve, rel_retrieve_params, ent_retrieve_params, tensor_dict_map, model_params_generator)
extra_args_formatter = lambda args: '' max_ent_len_retrieve = lambda train_doc_store, dev_doc_store, args: 1 max_rel_len_retrieve = lambda train_doc_store, dev_doc_store, args: 520 rel_retrieve_params = {'replacement': (ENT_1, ENT_2), 'truncate': True} ent_retrieve_params = lambda args: {'neighb_size': 0} tensor_dict_map = { 'rel_seq': 'rel_seq', 'ent_seq': 'ent_seq', 'ent_len': 'ent_len', 'rel_len': 'rel_len', 'seq_len': 'seq_len', 'target_rel': 'target_rel' } run_model(visible_device_list=visible_device_list, visible_devices=visible_devices, memory_fraction=memory_fraction, model_name=model_name, extra_parser_args_adder=extra_parser_args_adder, extra_args_formatter=extra_args_formatter, max_ent_len_retrieve=max_ent_len_retrieve, max_rel_len_retrieve=max_rel_len_retrieve, rel_retrieve_params=rel_retrieve_params, ent_retrieve_params=ent_retrieve_params, tensor_dict_map=tensor_dict_map, model_params_generator=model_params_generator, no_save=True)