|
@@ -166,7 +166,7 @@ To train this model, you simply need to specify the following:
|
|
bazel build -c opt --config=cuda inception/imagenet_train
|
|
bazel build -c opt --config=cuda inception/imagenet_train
|
|
|
|
|
|
# run it
|
|
# run it
|
|
-bazel-bin/inception/imagenet_train.py --num_gpus=1 --batch_size=32 --train_dir=/tmp/imagenet_train --data_dir=/tmp/imagenet_data
|
|
|
|
|
|
+bazel-bin/inception/imagenet_train --num_gpus=1 --batch_size=32 --train_dir=/tmp/imagenet_train --data_dir=/tmp/imagenet_data
|
|
```
|
|
```
|
|
|
|
|
|
The model reads in the ImageNet training data from `--data_dir`. If you followed
|
|
The model reads in the ImageNet training data from `--data_dir`. If you followed
|