Skip to content

Commit 8b35d7f

Browse files
author
Lee Yang
committed
minor edits
1 parent dca5538 commit 8b35d7f

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

examples/resnet/README.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,16 @@ Original Source: https://github.com/tensorflow/models/tree/master/official/visio
44

55
This code is based on the Image Classification model from the official [TensorFlow Models](https://github.com/tensorflow/models) repository. This example already supports different forms of distribution via the `DistributionStrategy` API, so there isn't much additional work to convert it to TensorFlowOnSpark.
66

7-
Notes:
7+
Notes:
88
- This example assumes that Spark, TensorFlow, and TensorFlowOnSpark are already installed.
99
- For simplicity, this just uses a single-node Spark Standalone installation.
1010

1111
#### Run the Single-Node Application
1212

13-
First, make sure that you can run the example per the [original instructions](https://github.com/tensorflow/models/tree/68c3c65596b8fc624be15aef6eac3dc8952cbf23/official/vision/image_classification). For now, we'll just use the CIFAR-10 dataset. After cloning the [tensorflow/models](https://github.com/tensorflow/models) repository and downloading the dataset, you should be able to run the training as follows:
13+
First, make sure that you can run the example per the [original instructions](https://github.com/tensorflow/models/tree/68c3c65596b8fc624be15aef6eac3dc8952cbf23/official/vision/image_classification). For now, we'll just use the CIFAR-10 dataset. After cloning the [tensorflow/models](https://github.com/tensorflow/models) repository (checking out the `v2.0` tag with `git checkout v2.0`), and downloading the dataset, you should be able to run the training as follows:
1414
```
15+
# Note: these instructions have been tested with the `v2.0` tag of tensorflow/models.
16+
1517
export TENSORFLOW_MODELS=/path/to/tensorflow/models
1618
export CIFAR_DATA=/path/to/cifar
1719
export PYTHONPATH=${PYTHONPATH}:${TENSORFLOW_MODELS}

tensorflowonspark/TFNode.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
logger = logging.getLogger(__name__)
2323

24+
2425
def hdfs_path(ctx, path):
2526
"""Convenience function to create a Tensorflow-compatible absolute HDFS path from relative paths
2627

0 commit comments

Comments
 (0)