From fbb055d55e7aa8e020a4e8f04dfb0986c23a921b Mon Sep 17 00:00:00 2001 From: massongit Date: Thu, 2 Nov 2017 17:12:36 +0900 Subject: [PATCH] Fix path of method of __init__.py --- deepcrf/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepcrf/__init__.py b/deepcrf/__init__.py index be0e30a..42a5a41 100644 --- a/deepcrf/__init__.py +++ b/deepcrf/__init__.py @@ -57,7 +57,7 @@ def cli(): @click.option('--efficient_gpu', type=int, default=1, help='efficient_gpu (if efficient_gpu == 1, it needs small GPU memory)') def train(train_file, **args): # load input_file - main.run(train_file, is_train=True, **args) + deepcrf.main.run(train_file, is_train=True, **args) @cli.command() @@ -107,7 +107,7 @@ def train(train_file, **args): @click.option('--use_cudnn', type=int, default=1, help='use_cudnn = 0 or 1') @click.option('--efficient_gpu', type=int, default=1, help='efficient_gpu (if efficient_gpu == 1, it needs small GPU memory)') def predict(input_file, **args): - main.run(input_file, is_train=False, **args) + deepcrf.main.run(input_file, is_train=False, **args) @cli.command() @@ -115,4 +115,4 @@ def predict(input_file, **args): @click.argument('predicted_file', type=click.Path(exists=True)) @click.option('--tag_type', type=str, default='BIOES', help='select from [BIO, BIOES]') def eval(gold_file, predicted_file, **args): - evaluate.run(gold_file, predicted_file, **args) + deepcrf.evaluate.run(gold_file, predicted_file, **args)