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