提交 10cb9424 编写于 作者: T Thuan Nguyen

remove enum reference from trainer

上级 6d58d6dc
...@@ -12,15 +12,13 @@ ...@@ -12,15 +12,13 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from enum import Enum
__all__ = [ __all__ = [
'Event', 'Event',
'Trainer', 'Trainer',
] ]
class Event(Enum): class Event(object):
BEGIN_EPOCH = 0 BEGIN_EPOCH = 0
END_EPOCH = 1 END_EPOCH = 1
BEGIN_STEP = 2 BEGIN_STEP = 2
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册