Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "08e62070f6dee22a8984f700147f581b8bb49fc5"
Commit f67822f5 authored by Katherine Wu's avatar Katherine Wu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 452369802
parent 4862ac84
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/modeling/fast_training/progressive/utils.py
View file @ f67822f5
...@@ -18,10 +18,10 @@ from absl import logging ...@@ -18,10 +18,10 @@ from absl import logging
import tensorflow as tf import tensorflow as tf
# pylint: disable=g-direct-tensorflow-import # pylint: disable=g-direct-tensorflow-import
from tensorflow.python.training.tracking import tracking from tensorflow.python.trackable import autotrackable
class VolatileTrackable(tracking.AutoTrackable): class VolatileTrackable(autotrackable.AutoTrackable):
"""A util class to keep Trackables that might change instances.""" """A util class to keep Trackables that might change instances."""
def __init__(self, **kwargs): def __init__(self, **kwargs):
......
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