From d46b4086140ce73ac9dc6e6cd16910503a8e5bec Mon Sep 17 00:00:00 2001 From: zhangchunle Date: Mon, 7 Jun 2021 10:27:41 +0800 Subject: [PATCH] fix undefined_all_variable (#32611) --- python/paddle/fluid/trainer_desc.py | 2 +- python/paddle/fluid/trainer_factory.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/trainer_desc.py b/python/paddle/fluid/trainer_desc.py index 989db9efea1..92a900e6c37 100644 --- a/python/paddle/fluid/trainer_desc.py +++ b/python/paddle/fluid/trainer_desc.py @@ -17,7 +17,7 @@ import sys import os __all__ = [ 'TrainerDesc', 'MultiTrainer', 'DistMultiTrainer', 'PipelineTrainer', - 'HeterXpuTrainer', 'HeterBoxWorker' + 'HeterXpuTrainer', 'HeterBoxTrainer' ] diff --git a/python/paddle/fluid/trainer_factory.py b/python/paddle/fluid/trainer_factory.py index 00dea8d1251..95379a34c22 100644 --- a/python/paddle/fluid/trainer_factory.py +++ b/python/paddle/fluid/trainer_factory.py @@ -27,7 +27,7 @@ from .device_worker import Hogwild, DownpourSGD, Section, DownpourSGDOPT from .framework import Variable from multiprocessing import Process, Manager -__all__ = ["TrainerFactory", "FetchHandler", "FetchHandlerMonitor"] +__all__ = ["TrainerFactory", "FetchHandlerMonitor"] class TrainerFactory(object): -- GitLab