diff --git a/PaddleCV/image_classification/utils/__init__.py b/PaddleCV/image_classification/utils/__init__.py index b03f382639faa9e9b90a9a344cd8145171292bad..4677e4535712c9f261bf18ba08ba6446d2db76d8 100644 --- a/PaddleCV/image_classification/utils/__init__.py +++ b/PaddleCV/image_classification/utils/__init__.py @@ -12,4 +12,4 @@ #See the License for the specific language governing permissions and #limitations under the License. from .optimizer import cosine_decay, lr_warmup, cosine_decay_with_warmup, exponential_decay_with_warmup, Optimizer, create_optimizer -from .utility import add_arguments, print_arguments, parse_args, check_gpu, check_args, init_model, save_model, create_data_loader, print_info, best_strategy_compiled, init_model, save_model, ExponentialMovingAverage +from .utility import add_arguments, print_arguments, parse_args, check_gpu, check_args, check_version, init_model, save_model, create_data_loader, print_info, best_strategy_compiled, init_model, save_model, ExponentialMovingAverage diff --git a/PaddleCV/image_classification/utils/utility.py b/PaddleCV/image_classification/utils/utility.py index 5586e905539950b3297584dc9a65d55d302e549a..61be68690656071b59464a1645ef88d04266b0d8 100644 --- a/PaddleCV/image_classification/utils/utility.py +++ b/PaddleCV/image_classification/utils/utility.py @@ -179,7 +179,7 @@ def check_version(): try: fluid.require_version('1.6.0') except Exception as e: - logger.error(err) + print(err) sys.exit(1)