From df8c6d287ff7d7f22ec117401f8c58be1f019b92 Mon Sep 17 00:00:00 2001 From: wuzewu Date: Wed, 11 Dec 2019 12:01:37 +0800 Subject: [PATCH] Update solver.py --- pdseg/solver.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pdseg/solver.py b/pdseg/solver.py index 8eea7400..17673747 100644 --- a/pdseg/solver.py +++ b/pdseg/solver.py @@ -18,7 +18,10 @@ import paddle.fluid as fluid import numpy as np import importlib from utils.config import cfg -from paddle.fluid.contrib.mixed_precision.decorator import OptimizerWithMixedPrecison, decorate, AutoMixedPrecisionLists +try: + from paddle.fluid.contrib.mixed_precision.decorator import OptimizerWithMixedPrecison, decorate, AutoMixedPrecisionLists +except: + from paddle.fluid.contrib.mixed_precision.decorator import OptimizerWithMixedPrecision, decorate, AutoMixedPrecisionLists class Solver(object): -- GitLab