Add Nano TensorFlow Training QuickStart Doc (#4591)
* Add Nano TensorFlow Training QuickStart Doc * fix typo * break training and inference to two files
This commit is contained in:
parent
0656804f56
commit
25982b0d16
4 changed files with 51 additions and 19 deletions
|
|
@ -91,4 +91,4 @@ model.compile(optimizer='adam',
|
||||||
model.fit(x_train, y_train, epochs=5, num_processes=4)
|
model.fit(x_train, y_train, epochs=5, num_processes=4)
|
||||||
```
|
```
|
||||||
|
|
||||||
For more details on the BigDL-Nano's PyTorch usage, please refer to the [TensorFlow](../QuickStart//tensorflow.md) page.
|
For more details on the BigDL-Nano's PyTorch usage, please refer to the [TensorFlow Training](../QuickStart//tensorflow.md) and [TensorFlow Inference](../QuickStart/tensorflow_inference.md) page.
|
||||||
|
|
|
||||||
|
|
@ -1,18 +0,0 @@
|
||||||
# Nano TensorFlow Overview
|
|
||||||
|
|
||||||
## TensorFlow Training
|
|
||||||
|
|
||||||
### Runtime Acceleration
|
|
||||||
intel-tensorflow, intel-openmp
|
|
||||||
|
|
||||||
### Optimized Layers
|
|
||||||
embedding
|
|
||||||
|
|
||||||
### Optimizers
|
|
||||||
SparseAdam
|
|
||||||
|
|
||||||
### Multi-Instance Training
|
|
||||||
|
|
||||||
## TensorFlow Inference
|
|
||||||
|
|
||||||
### Quantization
|
|
||||||
|
|
@ -0,0 +1,3 @@
|
||||||
|
# Nano TensorFlow Inference Overview
|
||||||
|
|
||||||
|
### Quantization
|
||||||
|
|
@ -0,0 +1,47 @@
|
||||||
|
# Nano TensorFlow Training Overview
|
||||||
|
|
||||||
|
BigDL-Nano can be used to accelerate TensorFlow Keras applications on training workloads. The optimizations in BigDL-Nano are delivered through BigDL-Nano's `Model` and `Sequential` classes, which have identical APIs with `tf.keras.Model` and `tf.keras.Sequential`. For most cases, you can just replace your `tf.keras.Model` to `bigdl.nano.tf.keras.Model` and `tf.keras.Sequential` to `bigdl.nano.tf.keras.Sequential` to benefits from BigDL-Nano.
|
||||||
|
|
||||||
|
We will briefly describe here the major features in BigDL-Nano for TensorFlow training. You can find complete examples here [links to be added]().
|
||||||
|
|
||||||
|
### Best Known Configurations
|
||||||
|
When you install BigDL-Nano by `pip install bigdl-nano[tensorflow]`, intel-tensorflow will be installed in your environment, which has intel's oneDNN optimizations enabled by default; and when you run `source bigdl-nano-init`, it will export a few environment variables, such as OMP_NUM_THREADS and KMP_AFFINITY, according to your current hardware. Empirically, these environment variables work best for most TensorFlow applications. After setting these environment variables, you can just run your applications as usual (`python app.py`) and no additional changes are required.
|
||||||
|
|
||||||
|
### Multi-Instance Training
|
||||||
|
|
||||||
|
When training on a server with dozens of CPU cores, it is often beneficial to use multiple training instances in a data-parallel fashion to make full use of the CPU cores. However, naively using TensorFlow's `MultiWorkerMirroredStrategy` can cause conflict in CPU cores and often cannot provide performance benefits.
|
||||||
|
|
||||||
|
BigDL-Nano makes it very easy to conduct multi-instance training correctly. You can just set the `num_processes` parameter in the `fit` method in your `Model` or `Sequential` object and BigDL-Nano will launch the specific number of processes to perform data-parallel training. Each process will be automatically pinned to a different subset of CPU cores to avoid conflict and maximize training throughput.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import tensorflow as tf
|
||||||
|
from tensorflow.keras import layers
|
||||||
|
from bigdl.nano.tf.keras import Sequential
|
||||||
|
|
||||||
|
model = Sequential([
|
||||||
|
layers.Rescaling(1. / 255, input_shape=(img_height, img_width, 3)),
|
||||||
|
layers.Conv2D(16, 3, padding='same', activation='relu'),
|
||||||
|
layers.MaxPooling2D(),
|
||||||
|
layers.Conv2D(32, 3, padding='same', activation='relu'),
|
||||||
|
layers.MaxPooling2D(),
|
||||||
|
layers.Conv2D(64, 3, padding='same', activation='relu'),
|
||||||
|
layers.MaxPooling2D(),
|
||||||
|
layers.Flatten(),
|
||||||
|
layers.Dense(128, activation='relu'),
|
||||||
|
layers.Dense(num_classes)
|
||||||
|
])
|
||||||
|
|
||||||
|
model.compile(optimizer='adam',
|
||||||
|
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
|
||||||
|
metrics=['accuracy'])
|
||||||
|
|
||||||
|
model.fit(train_ds, epochs=3, validation_data=val_ds, num_processes=2)
|
||||||
|
```
|
||||||
|
|
||||||
|
Note that, different from the conventions in PyTorch, the effective batch size will not change in TensorFlow multi-instance training, which means it is still the batch size you specify in your dataset. This is because TensorFlow's `MultiWorkerMirroredStrategy` will try to split the batch into multiple sub-batches for different workers. We chose this behavior to match the semantics of TensorFlow distributed training.
|
||||||
|
|
||||||
|
When you do want to increase your effective batch_size, you can do so by directly changing it in your dataset definition and you may also want to gradually increase the learning rate linearly to the batch_size, as described in the [Facebook paper](https://arxiv.org/abs/1706.02677).
|
||||||
|
|
||||||
|
## TensorFlow Inference
|
||||||
|
|
||||||
|
### Quantization
|
||||||
Loading…
Reference in a new issue