From c098364e2008c3ea86a96c3cc5f82712ebbbaaf1 Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Wed, 17 Jun 2020 17:25:03 +0800 Subject: [PATCH] update train.py --- dygraph/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dygraph/train.py b/dygraph/train.py index a776cb0b..17ba2d68 100644 --- a/dygraph/train.py +++ b/dygraph/train.py @@ -21,7 +21,7 @@ import paddle.fluid as fluid from paddle.fluid.dygraph.parallel import ParallelEnv from paddle.fluid.io import DataLoader -from datasets import OpticDiscSeg, Dataset +from datasets import OpticDiscSeg import transforms as T import models import utils.logging as logging -- GitLab