"megatron/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "d1f5776a53bbb5adac54d28bb5f216500578d36c"
Commit c60168f0 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix train_utils import

PiperOrigin-RevId: 326947485
parent 3b74fd29
......@@ -23,7 +23,7 @@ from absl import logging
import orbit
import tensorflow as tf
from official.common import train_utils
from official.core import train_utils
from official.core import base_task
from official.modeling.hyperparams import config_definitions
......
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