def init_fn(sess): start_type_generic = { 'nli': 'cls', 'nli_ex': 'cls_ex', 'bert': 'bert', 'cold': 'cold' }[start_type] return init_fn_generic(sess, start_type_generic, start_model_path)
def init_fn(sess): return init_fn_generic(sess, "as_is", start_model_path)
def init_fn(sess): return init_fn_generic(sess, start_type, start_model_path)