Commit 1fdfd973 authored by Allen Wang's avatar Allen Wang Committed by A. Unique TensorFlower
Browse files

Move resnet_model.py into its own folder.

PiperOrigin-RevId: 300362532
parent 93fc3c9f
......@@ -24,10 +24,10 @@ import os
from absl import app
from absl import flags
import tensorflow as tf
import tensorflow.compat.v2 as tf
from official.vision.image_classification import imagenet_preprocessing
from official.vision.image_classification import resnet_model
from official.vision.image_classification.resnet import resnet_model
FLAGS = flags.FLAGS
......
......@@ -36,7 +36,7 @@ from official.utils.misc import keras_utils
from official.utils.misc import model_helpers
from official.vision.image_classification import common
from official.vision.image_classification import imagenet_preprocessing
from official.vision.image_classification import resnet_model
from official.vision.image_classification.resnet import resnet_model
def run(flags_obj):
......
......@@ -27,7 +27,7 @@ from official.staging.training import utils
from official.utils.flags import core as flags_core
from official.vision.image_classification import common
from official.vision.image_classification import imagenet_preprocessing
from official.vision.image_classification import resnet_model
from official.vision.image_classification.resnet import resnet_model
class ResnetRunnable(standard_runnable.StandardTrainable,
......
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