Skip to content

make /official a Python module #3557

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Mar 12, 2018
Merged

make /official a Python module #3557

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
0af6b9b
Adding logging utils
karmel Feb 27, 2018
c83533d
restore utils
yhliang2018 Mar 7, 2018
3da8215
delete old file
yhliang2018 Mar 7, 2018
5f12bff
update inputs and docstrings
yhliang2018 Mar 8, 2018
be8c1ba
make /official a python module
yhliang2018 Mar 10, 2018
c86fc74
remove /utils directory
yhliang2018 Mar 10, 2018
21f748c
Update readme for python path setting
yhliang2018 Mar 12, 2018
5ed95a1
Change readme texts
yhliang2018 Mar 12, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion official/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@ The master branch of the models are **in development**, and they target the [nig

If you are on a version of TensorFlow earlier than 1.4, please [update your installation](https://www.tensorflow.org/install/).


---

Below is a list of the models available.
Expand All @@ -22,3 +21,11 @@ Below is a list of the models available.
More models to come!

If you would like to make any fixes or improvements to the models, please [submit a pull request](https://github.com/tensorflow/models/compare).

---

Run the models.

The *Official Models* are made available as a Python module. To run the models and associated scripts, add the top-level ***/models*** folder to the Python path with the command: `export PYTHONPATH="$PYTHONPATH:/path/to/models"`

To make Official Models easier to use, we are planning to create a pip installable Official Models package. This is being tracked in [#917](https://github.com/tensorflow/models/issues/917).
Empty file added official/__init__.py
View file
Open in desktop
Empty file.
2 changes: 1 addition & 1 deletion official/mnist/mnist.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import sys

import tensorflow as tf
import dataset
from official.mnist import dataset


class Model(tf.keras.Model):
Expand Down
4 changes: 2 additions & 2 deletions official/mnist/mnist_eager.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@

import tensorflow as tf
import tensorflow.contrib.eager as tfe
import mnist
import dataset
from official.mnist import mnist
from official.mnist import dataset

FLAGS = None

Expand Down
4 changes: 2 additions & 2 deletions official/mnist/mnist_eager_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@
import tensorflow as tf
import tensorflow.contrib.eager as tfe

import mnist
import mnist_eager
from official.mnist import mnist
from official.mnist import mnist_eager


def device():
Expand Down
2 changes: 1 addition & 1 deletion official/mnist/mnist_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import tensorflow as tf
import time

import mnist
from official.mnist import mnist

BATCH_SIZE = 100

Expand Down
4 changes: 2 additions & 2 deletions official/mnist/mnist_tpu.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@
from __future__ import print_function

import tensorflow as tf
import dataset
import mnist
from official.mnist import dataset
from official.mnist import mnist

# Cloud TPU Cluster Resolver flags
tf.flags.DEFINE_string(
Expand Down
2 changes: 1 addition & 1 deletion official/resnet/cifar10_main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@

import tensorflow as tf

import resnet
from official.resnet import resnet

_HEIGHT = 32
_WIDTH = 32
Expand Down
2 changes: 1 addition & 1 deletion official/resnet/cifar10_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import numpy as np
import tensorflow as tf

import cifar10_main
from official.resnet import cifar10_main

tf.logging.set_verbosity(tf.logging.ERROR)

Expand Down
4 changes: 2 additions & 2 deletions official/resnet/imagenet_main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,8 @@

import tensorflow as tf

import resnet
import vgg_preprocessing
from official.resnet import resnet
from official.resnet import vgg_preprocessing

_DEFAULT_IMAGE_SIZE = 224
_NUM_CHANNELS = 3
Expand Down
2 changes: 1 addition & 1 deletion official/resnet/imagenet_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@

import tensorflow as tf

import imagenet_main
from official.resnet import imagenet_main

tf.logging.set_verbosity(tf.logging.ERROR)

Expand Down
2 changes: 1 addition & 1 deletion official/wide_deep/wide_deep_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@

import tensorflow as tf

import wide_deep
from official.wide_deep import wide_deep

tf.logging.set_verbosity(tf.logging.ERROR)

Expand Down