@@ -13,7 +13,7 @@ For more information about other types of models, please refer to this
13
13
Similar to the [ estimator implementation] ( ../../r1/resnet ) , the Keras
14
14
implementation has code for the ImageNet dataset. The ImageNet
15
15
version uses a ResNet50 model implemented in
16
- [ ` resnet_model.py ` ] ( ./resnet_model.py ) .
16
+ [ ` resnet_model.py ` ] ( ./resnet/ resnet_model.py ) .
17
17
18
18
Please make sure that you have the latest version of TensorFlow
19
19
installed and
@@ -36,14 +36,14 @@ provide a few options.
36
36
Once your dataset is ready, you can begin training the model as follows:
37
37
38
38
``` bash
39
- python resnet_imagenet_main.py
39
+ python resnet/ resnet_imagenet_main.py
40
40
```
41
41
42
42
Again, if you did not download the data to the default directory, specify the
43
43
location with the ` --data_dir ` flag:
44
44
45
45
``` bash
46
- python resnet_imagenet_main.py --data_dir=/path/to/imagenet
46
+ python resnet/ resnet_imagenet_main.py --data_dir=/path/to/imagenet
47
47
```
48
48
49
49
There are more flag options you can specify. Here are some examples:
@@ -62,7 +62,7 @@ For example, this is a typical command line to run with ImageNet data with
62
62
batch size 128 per GPU:
63
63
64
64
``` bash
65
- python -m resnet_imagenet_main \
65
+ python -m resnet/ resnet_imagenet_main.py \
66
66
--model_dir=/tmp/model_dir/something \
67
67
--num_gpus=2 \
68
68
--batch_size=128 \
@@ -120,7 +120,7 @@ From a GCE VM, you can run the following command to train ResNet for one epoch
120
120
on a v2-8 or v3-8 TPU:
121
121
122
122
``` bash
123
- python resnet_ctl_imagenet_main.py \
123
+ python resnet/ resnet_ctl_imagenet_main.py \
124
124
--tpu=$TPU_NAME \
125
125
--model_dir=$MODEL_DIR \
126
126
--data_dir=$DATA_DIR \
@@ -140,7 +140,7 @@ python resnet_ctl_imagenet_main.py \
140
140
To train the ResNet to convergence, run it for 90 epochs:
141
141
142
142
``` bash
143
- python resnet_ctl_imagenet_main.py \
143
+ python resnet/ resnet_ctl_imagenet_main.py \
144
144
--tpu=$TPU_NAME \
145
145
--model_dir=$MODEL_DIR \
146
146
--data_dir=$DATA_DIR \
0 commit comments