Skip to content

GitLab

Commit 6f91155b authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Use pure PY3 syntax 

PiperOrigin-RevId: 326731205
parent bd9c6c4e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
official/core/base_task.py
View file @ 6f91155b
...@@ -19,14 +19,12 @@ import functools ...@@ -19,14 +19,12 @@ import functools
from typing import Any, Callable, Optional from typing import Any, Callable, Optional
from absl import logging from absl import logging
import six
import tensorflow as tf import tensorflow as tf
from official.modeling.hyperparams import config_definitions as cfg from official.modeling.hyperparams import config_definitions as cfg
@six.add_metaclass(abc.ABCMeta) class Task(tf.Module, metaclass=abc.ABCMeta):
class Task(tf.Module):
"""A single-replica view of training procedure. """A single-replica view of training procedure.
Tasks provide artifacts for training/evalution procedures, including Tasks provide artifacts for training/evalution procedures, including
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment