Note
Go to the end to download the full example code
Build Vision Transformers for Akida
The Vision Transformer, or ViT, is a model for image classification that employs a Transformer-like architecture over patches of the image. An image is split into fixed-size patches, each of them are then linearly embedded, position embeddings are added, and the resulting sequence of vectors are fed to a standard Transformer encoder. Please refer to https://arxiv.org/abs/2010.11929 for further details.
Akida 2.0 now supports patch and position embeddings, and the encoder block in hardware. This tutorial explains how to build an optimized ViT using Akida models python API for Akida 2.0 hardware.
1. Model selection
There are many variants of ViT. The choice of the model is typically influenced by the tradeoff among architecture size, accuracy, inference speed, and training capabilities.
The following table shows few variants of commonly used ViT:
Architecture |
Original accuracy |
#Params |
Architecture |
|---|---|---|---|
ViT Base |
79.90% |
86M |
12 heads, 12 blocks, hidden size 768 |
ViT Tiny |
75.48% |
5.8M |
3 heads, 12 blocks, hidden size 192 |
DeiT-dist Tiny |
74.17% |
5.8M |
3 heads, 12 blocks, hidden size 192 |
Note
The Vision Transformers support has been introduced in Akida 2.0.
The Akida model zoo provides tiny ViT architectures that are optimized to run on Akida hardware:
Both architectures have been modified so that their layers can be quantized to integer only operations.
2. Model optimization for Akida hardware
ViT has many encoder blocks that perform self-attention to process visual data. Each encoder block consists of many different layers. To optimally run ViT at the edge using Akida requires transforming this encoder block in the following way:
replace LayerNormalization with LayerMadNormalization,
replace the last LayerNormalization previous to the classification head with a BatchNormalization,
replace Softmax operation in Attention with a shiftmax operation.
Note
Sections below show different ways to train a ViT for Akida which uses the above transformations.
3. Model Training
Akida accelerates ViT model that has the transformation mentioned in Section 2. Training a ViT that optimally runs on Akida can be made possible in the following two ways:
3.1 Option 1: Training a ViT (original) model first and then transforming each layer incrementally
First, train a ViT (original) model on a custom dataset until satisfactory accuracy. It is then possible to transform this model into an Akida optimized one as per Section 2. The layers mentioned in Section 2 are functionally equivalent to each of the layers present in the original model.
Note
To overcome the accuracy drop from the original when transforming the model as per Section 2, it is recommended to replace the original layers all at once and to fine-tune afterwards.
The example below shows the transformation of ViT (tiny) into an optimized model that can run on the Akida hardware.
The akida_models python package provides a Command Line Interface (CLI) to transform vit_ti16 and deit_ti16 model architectures and fine-tune them respectively.
$ akida_models create vit_ti16 -h
usage: akida_models create vit_ti16 [-h] [-c CLASSES] [-bw BASE_WEIGHTS] [--norm {LN,GN1,BN,LMN}]
[--last_norm {LN,BN}] [--softmax {softmax,softmax2}]
[--act {GeLU,ReLU8,swish}] [-i {224,384}]
optional arguments:
-h, --help show this help message and exit
-c CLASSES, --classes CLASSES
The number of classes, by default 1000.
-bw BASE_WEIGHTS, --base_weights BASE_WEIGHTS
Optional keras weights to load in the model, by default None.
--norm {LN,GN1,BN,LMN}
Replace normalization in model with a custom function, by default LN
--last_norm {LN,BN} Replace last normalization in model with a custom function, by default LN
--softmax {softmax,softmax2}
Replace softmax operation in model with custom function, by default softmax
--act {GeLU,ReLU8,swish}
Replace activation function in model with custom function, by default GeLU
-i {224,384}, --image_size {224,384}
The square input image size
The following shows the transformation of a vit_ti16 model architecture which was trained on ImageNet. The same methods can be applied for other datasets.
# download the pre-trained weights
wget https://data.brainchip.com/models/AkidaV2/vit/vit_ti16_224.h5
# transformations: replace layer normalization with mad norm layer, last layer normalization
# with batch normalization, GeLU layer with ReLU and softmax with shiftmax layer
akida_models create -s vit_ti16_transformed.h5 vit_ti16 --norm LMN --last_norm BN --act ReLU8 \
--softmax softmax2 -bw vit_ti16_224.h5
# fine-tuning
imagenet_train tune -m vit_ti16_transformed.h5 -e 30 --optim Adam --lr_policy cosine_decay \
-lr 6e-5 -s vit_ti16_transformed.h5
The above transformation generates a ViT model that is optimized to run efficiently on Akida hardware. Similar steps can also be applied to deit_ti16. The table below highlights the accuracy of the original and transformed models.
Architecture |
Original accuracy |
Transformed accuracy |
|---|---|---|
ViT |
75.48% |
74.25% |
DeiT-dist |
74.17% |
75.03% |
Note
The models obtained above have floating point weights and are ready to be quantized. See Section 4.
3.2 Option 2: Transfer Learning using Pre-trained transformed model
The Akida models python package has APIs for ViTs which provides pre-trained models for vit_ti16 and deit_ti16. These models can be used for Transfer Learning on a custom dataset. Since the above models are already transformed, no further transformation is required.
Visit our Transfer Learning Example to learn more about Transfer Learning using the Akida models python package. The following code snippet downloads a pre-trained model that can be used for Transfer Learning.
# The following is the API download the vit_t16 model trained on ImageNet dataset
from akida_models import fetch_file
from akida_models.model_io import load_model
# Retrieve the float model with pretrained weights and load it
model_file = fetch_file(
fname="bc_vit_ti16_224.h5",
origin="https://data.brainchip.com/models/AkidaV2/vit/bc_vit_ti16_224.h5",
cache_subdir='models/akidanet_imagenet')
model_keras = load_model(model_file)
model_keras.summary()
Downloading data from https://data.brainchip.com/models/AkidaV2/vit/bc_vit_ti16_224.h5.
0/23695632 [..............................] - ETA: 0s
114688/23695632 [..............................] - ETA: 10s
958464/23695632 [>.............................] - ETA: 2s
3170304/23695632 [===>..........................] - ETA: 0s
6135808/23695632 [======>.......................] - ETA: 0s
8658944/23695632 [=========>....................] - ETA: 0s
11223040/23695632 [=============>................] - ETA: 0s
13598720/23695632 [================>.............] - ETA: 0s
15925248/23695632 [===================>..........] - ETA: 0s
18464768/23695632 [======================>.......] - ETA: 0s
20987904/23695632 [=========================>....] - ETA: 0s
23695632/23695632 [==============================] - 1s 0us/step
Download complete.
/usr/local/lib/python3.8/dist-packages/keras/initializers/initializers.py:120: UserWarning: The initializer TruncatedNormal is unseeded and being called multiple times, which will return identical values each time (even if the initializer is unseeded). Please update your code to provide a seed to the initializer, or avoid using the same initalizer instance more than once.
warnings.warn(
Model: "vit-tiny"
__________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
==================================================================================================
input (InputLayer) [(None, 224, 224, 3 0 []
)]
Rescale (Rescaling) (None, 224, 224, 3) 0 ['input[0][0]']
Embedding (Conv2D) (None, 14, 14, 192) 147648 ['Rescale[0][0]']
reshape (Reshape) (None, 196, 192) 0 ['Embedding[0][0]']
ClassToken (ClassToken) (None, 197, 192) 192 ['reshape[0][0]']
Transformer/PosEmbed (AddPosit (None, 197, 192) 37824 ['ClassToken[0][0]']
ionEmbs)
Transformer/EncoderBlock_0/Lay (None, 197, 192) 384 ['Transformer/PosEmbed[0][0]']
erNorm_0 (LayerMadNormalizatio
n)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_0/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_0/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_0/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_0/add (None, 197, 192) 0 ['dropout[0][0]',
_1 (Add) 'Transformer/PosEmbed[0][0]']
Transformer/EncoderBlock_0/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_0/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_0/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_0/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_0/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_0/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_1 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_0/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_0/Mlp (None, 197, 192) 147648 ['dropout_1[0][0]']
Block/Dense_1 (Dense)
dropout_2 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_0/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_0/add (None, 197, 192) 0 ['Transformer/EncoderBlock_0/add_
_2 (Add) 1[0][0]',
'dropout_2[0][0]']
Transformer/EncoderBlock_1/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_0/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_1/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_1/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_1/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_3 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_1/add (None, 197, 192) 0 ['dropout_3[0][0]',
_1 (Add) 'Transformer/EncoderBlock_0/add_
2[0][0]']
Transformer/EncoderBlock_1/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_1/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_1/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_1/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_1/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_1/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_4 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_1/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_1/Mlp (None, 197, 192) 147648 ['dropout_4[0][0]']
Block/Dense_1 (Dense)
dropout_5 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_1/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_1/add (None, 197, 192) 0 ['Transformer/EncoderBlock_1/add_
_2 (Add) 1[0][0]',
'dropout_5[0][0]']
Transformer/EncoderBlock_2/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_1/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_2/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_2/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_2/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_6 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_2/add (None, 197, 192) 0 ['dropout_6[0][0]',
_1 (Add) 'Transformer/EncoderBlock_1/add_
2[0][0]']
Transformer/EncoderBlock_2/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_2/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_2/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_2/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_2/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_2/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_7 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_2/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_2/Mlp (None, 197, 192) 147648 ['dropout_7[0][0]']
Block/Dense_1 (Dense)
dropout_8 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_2/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_2/add (None, 197, 192) 0 ['Transformer/EncoderBlock_2/add_
_2 (Add) 1[0][0]',
'dropout_8[0][0]']
Transformer/EncoderBlock_3/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_2/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_3/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_3/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_3/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_9 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_3/add (None, 197, 192) 0 ['dropout_9[0][0]',
_1 (Add) 'Transformer/EncoderBlock_2/add_
2[0][0]']
Transformer/EncoderBlock_3/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_3/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_3/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_3/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_3/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_3/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_10 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_3/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_3/Mlp (None, 197, 192) 147648 ['dropout_10[0][0]']
Block/Dense_1 (Dense)
dropout_11 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_3/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_3/add (None, 197, 192) 0 ['Transformer/EncoderBlock_3/add_
_2 (Add) 1[0][0]',
'dropout_11[0][0]']
Transformer/EncoderBlock_4/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_3/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_4/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_4/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_4/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_12 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_4/add (None, 197, 192) 0 ['dropout_12[0][0]',
_1 (Add) 'Transformer/EncoderBlock_3/add_
2[0][0]']
Transformer/EncoderBlock_4/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_4/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_4/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_4/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_4/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_4/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_13 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_4/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_4/Mlp (None, 197, 192) 147648 ['dropout_13[0][0]']
Block/Dense_1 (Dense)
dropout_14 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_4/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_4/add (None, 197, 192) 0 ['Transformer/EncoderBlock_4/add_
_2 (Add) 1[0][0]',
'dropout_14[0][0]']
Transformer/EncoderBlock_5/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_4/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_5/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_5/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_5/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_15 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_5/add (None, 197, 192) 0 ['dropout_15[0][0]',
_1 (Add) 'Transformer/EncoderBlock_4/add_
2[0][0]']
Transformer/EncoderBlock_5/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_5/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_5/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_5/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_5/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_5/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_16 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_5/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_5/Mlp (None, 197, 192) 147648 ['dropout_16[0][0]']
Block/Dense_1 (Dense)
dropout_17 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_5/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_5/add (None, 197, 192) 0 ['Transformer/EncoderBlock_5/add_
_2 (Add) 1[0][0]',
'dropout_17[0][0]']
Transformer/EncoderBlock_6/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_5/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_6/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_6/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_6/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_18 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_6/add (None, 197, 192) 0 ['dropout_18[0][0]',
_1 (Add) 'Transformer/EncoderBlock_5/add_
2[0][0]']
Transformer/EncoderBlock_6/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_6/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_6/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_6/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_6/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_6/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_19 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_6/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_6/Mlp (None, 197, 192) 147648 ['dropout_19[0][0]']
Block/Dense_1 (Dense)
dropout_20 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_6/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_6/add (None, 197, 192) 0 ['Transformer/EncoderBlock_6/add_
_2 (Add) 1[0][0]',
'dropout_20[0][0]']
Transformer/EncoderBlock_7/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_6/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_7/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_7/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_7/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_21 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_7/add (None, 197, 192) 0 ['dropout_21[0][0]',
_1 (Add) 'Transformer/EncoderBlock_6/add_
2[0][0]']
Transformer/EncoderBlock_7/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_7/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_7/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_7/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_7/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_7/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_22 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_7/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_7/Mlp (None, 197, 192) 147648 ['dropout_22[0][0]']
Block/Dense_1 (Dense)
dropout_23 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_7/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_7/add (None, 197, 192) 0 ['Transformer/EncoderBlock_7/add_
_2 (Add) 1[0][0]',
'dropout_23[0][0]']
Transformer/EncoderBlock_8/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_7/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_8/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_8/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_8/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_24 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_8/add (None, 197, 192) 0 ['dropout_24[0][0]',
_1 (Add) 'Transformer/EncoderBlock_7/add_
2[0][0]']
Transformer/EncoderBlock_8/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_8/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_8/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_8/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_8/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_8/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_25 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_8/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_8/Mlp (None, 197, 192) 147648 ['dropout_25[0][0]']
Block/Dense_1 (Dense)
dropout_26 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_8/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_8/add (None, 197, 192) 0 ['Transformer/EncoderBlock_8/add_
_2 (Add) 1[0][0]',
'dropout_26[0][0]']
Transformer/EncoderBlock_9/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_8/add_
erNorm_0 (LayerMadNormalizatio 2[0][0]']
n)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (Dense)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (Dense)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (Dense)
Transformer/EncoderBlock_9/Mul ((None, 197, 192), 0 ['Transformer/EncoderBlock_9/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (Attention) )) 0][0]',
'Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37056 ['Transformer/EncoderBlock_9/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (Dense) ion[0][0]']
dropout_27 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_9/add (None, 197, 192) 0 ['dropout_27[0][0]',
_1 (Add) 'Transformer/EncoderBlock_8/add_
2[0][0]']
Transformer/EncoderBlock_9/Lay (None, 197, 192) 384 ['Transformer/EncoderBlock_9/add_
erNorm_2 (LayerMadNormalizatio 1[0][0]']
n)
Transformer/EncoderBlock_9/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_9/Laye
Block/Dense_0 (Dense) rNorm_2[0][0]']
Transformer/EncoderBlock_9/Mlp (None, 197, 768) 0 ['Transformer/EncoderBlock_9/MlpB
Block/activation (ReLU) lock/Dense_0[0][0]']
dropout_28 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_9/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_9/Mlp (None, 197, 192) 147648 ['dropout_28[0][0]']
Block/Dense_1 (Dense)
dropout_29 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_9/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_9/add (None, 197, 192) 0 ['Transformer/EncoderBlock_9/add_
_2 (Add) 1[0][0]',
'dropout_29[0][0]']
Transformer/EncoderBlock_10/La (None, 197, 192) 384 ['Transformer/EncoderBlock_9/add_
yerNorm_0 (LayerMadNormalizati 2[0][0]']
on)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/q erNorm_0[0][0]']
uery (Dense)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/k erNorm_0[0][0]']
ey (Dense)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/v erNorm_0[0][0]']
alue (Dense)
Transformer/EncoderBlock_10/Mu ((None, 197, 192), 0 ['Transformer/EncoderBlock_10/Mul
ltiHeadDotProductAttention_1/a (None, 3, 197, 197 tiHeadDotProductAttention_1/query
ttention (Attention) )) [0][0]',
'Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/key[0
][0]',
'Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/value
[0][0]']
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_10/Mul
ltiHeadDotProductAttention_1/o tiHeadDotProductAttention_1/atten
ut (Dense) tion[0][0]']
dropout_30 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/out[0
][0]']
Transformer/EncoderBlock_10/ad (None, 197, 192) 0 ['dropout_30[0][0]',
d_1 (Add) 'Transformer/EncoderBlock_9/add_
2[0][0]']
Transformer/EncoderBlock_10/La (None, 197, 192) 384 ['Transformer/EncoderBlock_10/add
yerNorm_2 (LayerMadNormalizati _1[0][0]']
on)
Transformer/EncoderBlock_10/Ml (None, 197, 768) 148224 ['Transformer/EncoderBlock_10/Lay
pBlock/Dense_0 (Dense) erNorm_2[0][0]']
Transformer/EncoderBlock_10/Ml (None, 197, 768) 0 ['Transformer/EncoderBlock_10/Mlp
pBlock/activation (ReLU) Block/Dense_0[0][0]']
dropout_31 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_10/Mlp
Block/activation[0][0]']
Transformer/EncoderBlock_10/Ml (None, 197, 192) 147648 ['dropout_31[0][0]']
pBlock/Dense_1 (Dense)
dropout_32 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_10/Mlp
Block/Dense_1[0][0]']
Transformer/EncoderBlock_10/ad (None, 197, 192) 0 ['Transformer/EncoderBlock_10/add
d_2 (Add) _1[0][0]',
'dropout_32[0][0]']
Transformer/EncoderBlock_11/La (None, 197, 192) 384 ['Transformer/EncoderBlock_10/add
yerNorm_0 (LayerMadNormalizati _2[0][0]']
on)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/q erNorm_0[0][0]']
uery (Dense)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/k erNorm_0[0][0]']
ey (Dense)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/v erNorm_0[0][0]']
alue (Dense)
Transformer/EncoderBlock_11/Mu ((None, 197, 192), 0 ['Transformer/EncoderBlock_11/Mul
ltiHeadDotProductAttention_1/a (None, 3, 197, 197 tiHeadDotProductAttention_1/query
ttention (Attention) )) [0][0]',
'Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/key[0
][0]',
'Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/value
[0][0]']
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37056 ['Transformer/EncoderBlock_11/Mul
ltiHeadDotProductAttention_1/o tiHeadDotProductAttention_1/atten
ut (Dense) tion[0][0]']
dropout_33 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/out[0
][0]']
Transformer/EncoderBlock_11/ad (None, 197, 192) 0 ['dropout_33[0][0]',
d_1 (Add) 'Transformer/EncoderBlock_10/add
_2[0][0]']
Transformer/EncoderBlock_11/La (None, 197, 192) 384 ['Transformer/EncoderBlock_11/add
yerNorm_2 (LayerMadNormalizati _1[0][0]']
on)
Transformer/EncoderBlock_11/Ml (None, 197, 768) 148224 ['Transformer/EncoderBlock_11/Lay
pBlock/Dense_0 (Dense) erNorm_2[0][0]']
Transformer/EncoderBlock_11/Ml (None, 197, 768) 0 ['Transformer/EncoderBlock_11/Mlp
pBlock/activation (ReLU) Block/Dense_0[0][0]']
dropout_34 (Dropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_11/Mlp
Block/activation[0][0]']
Transformer/EncoderBlock_11/Ml (None, 197, 192) 147648 ['dropout_34[0][0]']
pBlock/Dense_1 (Dense)
dropout_35 (Dropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_11/Mlp
Block/Dense_1[0][0]']
Transformer/EncoderBlock_11/ad (None, 197, 192) 0 ['Transformer/EncoderBlock_11/add
d_2 (Add) _1[0][0]',
'dropout_35[0][0]']
Transformer/EncoderNorm (Batch (None, 197, 192) 768 ['Transformer/EncoderBlock_11/add
Normalization) _2[0][0]']
ExtractToken (ExtractToken) (None, 192) 0 ['Transformer/EncoderNorm[0][0]']
Head (Dense) (None, 1000) 193000 ['ExtractToken[0][0]']
==================================================================================================
Total params: 5,717,800
Trainable params: 5,717,416
Non-trainable params: 384
__________________________________________________________________________________________________
Note
The models in Section 3 have floating point weights. Once the desired accuracy is obtained, these models should go through quantization before converting to Akida.
4. Model quantization
Akida 2.0 hardware adds efficient processing of 8-bit weights and activations for Vision Transformer models. This requires models in Section 3 to be quantized to 8-bit integer numbers. This means both weights and activation outputs become 8-bit integer numbers. This results in a smaller model with minimal to no drop in accuracy and achieves improvements in latency and power when running on Akida hardware.
Quantization of ViT models can be done using QuantizeML python package using either Post Training Quantization (PTQ) or Quantization Aware Training (QAT) methods. The following section shows quantization an example, quantization of vit_ti16 trained on ImageNet dataset.
4.1 Post-Training Quantization
Using QuantizeML python package, ViT model can be quantized to 8-bit integer numbers (both weights and activation outputs). PTQ requires calibration (ideally using reference data) which helps to determine optimal quantization ranges. To learn more about PTQ, refer to Advanced QuantizeML tutorial.
# Using QuantizeML to perform quantization
from quantizeml.models import quantize, QuantizationParams
# Define the quantization parameters.
qparams = QuantizationParams(weight_bits=8, activation_bits=8)
# Quantize the model defined in Section 3.2
model_quantized = quantize(model_keras, qparams=qparams)
model_quantized.summary()
/usr/local/lib/python3.8/dist-packages/quantizeml/models/quantize.py:466: UserWarning: Quantizing per-axis with random calibration samples is not accurate. Set QuantizationParams.per_tensor_activations=True when calibrating with random samples.
warnings.warn("Quantizing per-axis with random calibration samples is not accurate.\
1/1024 [..............................] - ETA: 1:10:34
6/1024 [..............................] - ETA: 12s
11/1024 [..............................] - ETA: 12s
16/1024 [..............................] - ETA: 12s
21/1024 [..............................] - ETA: 12s
26/1024 [..............................] - ETA: 11s
30/1024 [..............................] - ETA: 12s
35/1024 [>.............................] - ETA: 11s
40/1024 [>.............................] - ETA: 11s
45/1024 [>.............................] - ETA: 11s
50/1024 [>.............................] - ETA: 11s
55/1024 [>.............................] - ETA: 11s
60/1024 [>.............................] - ETA: 11s
65/1024 [>.............................] - ETA: 11s
70/1024 [=>............................] - ETA: 11s
75/1024 [=>............................] - ETA: 11s
80/1024 [=>............................] - ETA: 11s
85/1024 [=>............................] - ETA: 11s
90/1024 [=>............................] - ETA: 11s
95/1024 [=>............................] - ETA: 11s
100/1024 [=>............................] - ETA: 11s
105/1024 [==>...........................] - ETA: 11s
110/1024 [==>...........................] - ETA: 11s
115/1024 [==>...........................] - ETA: 10s
120/1024 [==>...........................] - ETA: 10s
125/1024 [==>...........................] - ETA: 10s
130/1024 [==>...........................] - ETA: 10s
135/1024 [==>...........................] - ETA: 10s
140/1024 [===>..........................] - ETA: 10s
145/1024 [===>..........................] - ETA: 10s
150/1024 [===>..........................] - ETA: 10s
155/1024 [===>..........................] - ETA: 10s
160/1024 [===>..........................] - ETA: 10s
165/1024 [===>..........................] - ETA: 10s
170/1024 [===>..........................] - ETA: 10s
175/1024 [====>.........................] - ETA: 10s
180/1024 [====>.........................] - ETA: 10s
185/1024 [====>.........................] - ETA: 10s
190/1024 [====>.........................] - ETA: 10s
195/1024 [====>.........................] - ETA: 9s
200/1024 [====>.........................] - ETA: 9s
205/1024 [=====>........................] - ETA: 9s
210/1024 [=====>........................] - ETA: 9s
215/1024 [=====>........................] - ETA: 9s
220/1024 [=====>........................] - ETA: 9s
225/1024 [=====>........................] - ETA: 9s
230/1024 [=====>........................] - ETA: 9s
235/1024 [=====>........................] - ETA: 9s
240/1024 [======>.......................] - ETA: 9s
245/1024 [======>.......................] - ETA: 9s
250/1024 [======>.......................] - ETA: 9s
255/1024 [======>.......................] - ETA: 9s
260/1024 [======>.......................] - ETA: 9s
265/1024 [======>.......................] - ETA: 9s
270/1024 [======>.......................] - ETA: 9s
275/1024 [=======>......................] - ETA: 9s
280/1024 [=======>......................] - ETA: 8s
285/1024 [=======>......................] - ETA: 8s
290/1024 [=======>......................] - ETA: 8s
295/1024 [=======>......................] - ETA: 8s
300/1024 [=======>......................] - ETA: 8s
305/1024 [=======>......................] - ETA: 8s
310/1024 [========>.....................] - ETA: 8s
315/1024 [========>.....................] - ETA: 8s
320/1024 [========>.....................] - ETA: 8s
325/1024 [========>.....................] - ETA: 8s
330/1024 [========>.....................] - ETA: 8s
335/1024 [========>.....................] - ETA: 8s
340/1024 [========>.....................] - ETA: 8s
345/1024 [=========>....................] - ETA: 8s
350/1024 [=========>....................] - ETA: 8s
355/1024 [=========>....................] - ETA: 8s
360/1024 [=========>....................] - ETA: 7s
365/1024 [=========>....................] - ETA: 7s
370/1024 [=========>....................] - ETA: 7s
375/1024 [=========>....................] - ETA: 7s
380/1024 [==========>...................] - ETA: 7s
385/1024 [==========>...................] - ETA: 7s
390/1024 [==========>...................] - ETA: 7s
395/1024 [==========>...................] - ETA: 7s
400/1024 [==========>...................] - ETA: 7s
405/1024 [==========>...................] - ETA: 7s
410/1024 [===========>..................] - ETA: 7s
415/1024 [===========>..................] - ETA: 7s
420/1024 [===========>..................] - ETA: 7s
425/1024 [===========>..................] - ETA: 7s
430/1024 [===========>..................] - ETA: 7s
435/1024 [===========>..................] - ETA: 7s
440/1024 [===========>..................] - ETA: 7s
445/1024 [============>.................] - ETA: 6s
450/1024 [============>.................] - ETA: 6s
455/1024 [============>.................] - ETA: 6s
460/1024 [============>.................] - ETA: 6s
465/1024 [============>.................] - ETA: 6s
470/1024 [============>.................] - ETA: 6s
475/1024 [============>.................] - ETA: 6s
480/1024 [=============>................] - ETA: 6s
485/1024 [=============>................] - ETA: 6s
490/1024 [=============>................] - ETA: 6s
495/1024 [=============>................] - ETA: 6s
500/1024 [=============>................] - ETA: 6s
505/1024 [=============>................] - ETA: 6s
510/1024 [=============>................] - ETA: 6s
515/1024 [==============>...............] - ETA: 6s
520/1024 [==============>...............] - ETA: 6s
525/1024 [==============>...............] - ETA: 6s
530/1024 [==============>...............] - ETA: 5s
535/1024 [==============>...............] - ETA: 5s
540/1024 [==============>...............] - ETA: 5s
545/1024 [==============>...............] - ETA: 5s
550/1024 [===============>..............] - ETA: 5s
555/1024 [===============>..............] - ETA: 5s
560/1024 [===============>..............] - ETA: 5s
565/1024 [===============>..............] - ETA: 5s
570/1024 [===============>..............] - ETA: 5s
575/1024 [===============>..............] - ETA: 5s
580/1024 [===============>..............] - ETA: 5s
585/1024 [================>.............] - ETA: 5s
590/1024 [================>.............] - ETA: 5s
595/1024 [================>.............] - ETA: 5s
600/1024 [================>.............] - ETA: 5s
605/1024 [================>.............] - ETA: 5s
610/1024 [================>.............] - ETA: 4s
615/1024 [=================>............] - ETA: 4s
620/1024 [=================>............] - ETA: 4s
625/1024 [=================>............] - ETA: 4s
630/1024 [=================>............] - ETA: 4s
635/1024 [=================>............] - ETA: 4s
640/1024 [=================>............] - ETA: 4s
645/1024 [=================>............] - ETA: 4s
650/1024 [==================>...........] - ETA: 4s
655/1024 [==================>...........] - ETA: 4s
660/1024 [==================>...........] - ETA: 4s
665/1024 [==================>...........] - ETA: 4s
670/1024 [==================>...........] - ETA: 4s
675/1024 [==================>...........] - ETA: 4s
680/1024 [==================>...........] - ETA: 4s
685/1024 [===================>..........] - ETA: 4s
690/1024 [===================>..........] - ETA: 4s
695/1024 [===================>..........] - ETA: 3s
700/1024 [===================>..........] - ETA: 3s
705/1024 [===================>..........] - ETA: 3s
710/1024 [===================>..........] - ETA: 3s
715/1024 [===================>..........] - ETA: 3s
720/1024 [====================>.........] - ETA: 3s
725/1024 [====================>.........] - ETA: 3s
730/1024 [====================>.........] - ETA: 3s
735/1024 [====================>.........] - ETA: 3s
740/1024 [====================>.........] - ETA: 3s
745/1024 [====================>.........] - ETA: 3s
750/1024 [====================>.........] - ETA: 3s
755/1024 [=====================>........] - ETA: 3s
760/1024 [=====================>........] - ETA: 3s
765/1024 [=====================>........] - ETA: 3s
770/1024 [=====================>........] - ETA: 3s
775/1024 [=====================>........] - ETA: 2s
780/1024 [=====================>........] - ETA: 2s
785/1024 [=====================>........] - ETA: 2s
790/1024 [======================>.......] - ETA: 2s
795/1024 [======================>.......] - ETA: 2s
800/1024 [======================>.......] - ETA: 2s
805/1024 [======================>.......] - ETA: 2s
810/1024 [======================>.......] - ETA: 2s
815/1024 [======================>.......] - ETA: 2s
820/1024 [=======================>......] - ETA: 2s
825/1024 [=======================>......] - ETA: 2s
830/1024 [=======================>......] - ETA: 2s
835/1024 [=======================>......] - ETA: 2s
840/1024 [=======================>......] - ETA: 2s
845/1024 [=======================>......] - ETA: 2s
850/1024 [=======================>......] - ETA: 2s
855/1024 [========================>.....] - ETA: 2s
860/1024 [========================>.....] - ETA: 1s
865/1024 [========================>.....] - ETA: 1s
870/1024 [========================>.....] - ETA: 1s
875/1024 [========================>.....] - ETA: 1s
880/1024 [========================>.....] - ETA: 1s
885/1024 [========================>.....] - ETA: 1s
890/1024 [=========================>....] - ETA: 1s
895/1024 [=========================>....] - ETA: 1s
900/1024 [=========================>....] - ETA: 1s
905/1024 [=========================>....] - ETA: 1s
910/1024 [=========================>....] - ETA: 1s
915/1024 [=========================>....] - ETA: 1s
920/1024 [=========================>....] - ETA: 1s
925/1024 [==========================>...] - ETA: 1s
930/1024 [==========================>...] - ETA: 1s
935/1024 [==========================>...] - ETA: 1s
940/1024 [==========================>...] - ETA: 1s
945/1024 [==========================>...] - ETA: 0s
950/1024 [==========================>...] - ETA: 0s
955/1024 [==========================>...] - ETA: 0s
960/1024 [===========================>..] - ETA: 0s
965/1024 [===========================>..] - ETA: 0s
970/1024 [===========================>..] - ETA: 0s
975/1024 [===========================>..] - ETA: 0s
980/1024 [===========================>..] - ETA: 0s
985/1024 [===========================>..] - ETA: 0s
990/1024 [============================>.] - ETA: 0s
995/1024 [============================>.] - ETA: 0s
1000/1024 [============================>.] - ETA: 0s
1005/1024 [============================>.] - ETA: 0s
1010/1024 [============================>.] - ETA: 0s
1015/1024 [============================>.] - ETA: 0s
1020/1024 [============================>.] - ETA: 0s
1024/1024 [==============================] - 16s 12ms/step
Model: "vit-tiny"
__________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
==================================================================================================
input (InputLayer) [(None, 224, 224, 3 0 []
)]
Rescale (QuantizedRescaling) (None, 224, 224, 3) 0 ['input[0][0]']
Embedding (QuantizedConv2D) (None, 14, 14, 192) 147648 ['Rescale[0][0]']
reshape (QuantizedReshape) (None, 196, 192) 0 ['Embedding[0][0]']
ClassToken (QuantizedClassToke (None, 197, 192) 192 ['reshape[0][0]']
n)
Transformer/PosEmbed (Quantize (None, 197, 192) 38208 ['ClassToken[0][0]']
dAddPositionEmbs)
Transformer/EncoderBlock_0/Lay (None, 197, 192) 768 ['Transformer/PosEmbed[0][0]']
erNorm_0 (QuantizedLayerNormal
ization)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_0/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_0/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_0/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_0/add (None, 197, 192) 384 ['dropout[0][0]',
_1 (QuantizedAdd) 'Transformer/PosEmbed[0][0]']
Transformer/EncoderBlock_0/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_0/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_0/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_0/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_0/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_0/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_1 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_0/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_0/Mlp (None, 197, 192) 148032 ['dropout_1[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_2 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_0/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_0/add (None, 197, 192) 384 ['Transformer/EncoderBlock_0/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_2[0][0]']
Transformer/EncoderBlock_1/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_0/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_1/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_1/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_1/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_3 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_1/add (None, 197, 192) 384 ['dropout_3[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_0/add_
2[0][0]']
Transformer/EncoderBlock_1/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_1/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_1/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_1/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_1/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_1/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_4 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_1/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_1/Mlp (None, 197, 192) 148032 ['dropout_4[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_5 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_1/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_1/add (None, 197, 192) 384 ['Transformer/EncoderBlock_1/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_5[0][0]']
Transformer/EncoderBlock_2/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_1/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_2/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_2/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_2/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_6 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_2/add (None, 197, 192) 384 ['dropout_6[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_1/add_
2[0][0]']
Transformer/EncoderBlock_2/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_2/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_2/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_2/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_2/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_2/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_7 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_2/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_2/Mlp (None, 197, 192) 148032 ['dropout_7[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_8 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_2/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_2/add (None, 197, 192) 384 ['Transformer/EncoderBlock_2/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_8[0][0]']
Transformer/EncoderBlock_3/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_2/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_3/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_3/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_3/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_9 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_3/add (None, 197, 192) 384 ['dropout_9[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_2/add_
2[0][0]']
Transformer/EncoderBlock_3/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_3/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_3/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_3/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_3/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_3/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_10 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_3/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_3/Mlp (None, 197, 192) 148032 ['dropout_10[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_11 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_3/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_3/add (None, 197, 192) 384 ['Transformer/EncoderBlock_3/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_11[0][0]']
Transformer/EncoderBlock_4/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_3/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_4/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_4/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_4/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_12 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_4/add (None, 197, 192) 384 ['dropout_12[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_3/add_
2[0][0]']
Transformer/EncoderBlock_4/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_4/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_4/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_4/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_4/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_4/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_13 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_4/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_4/Mlp (None, 197, 192) 148032 ['dropout_13[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_14 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_4/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_4/add (None, 197, 192) 384 ['Transformer/EncoderBlock_4/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_14[0][0]']
Transformer/EncoderBlock_5/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_4/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_5/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_5/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_5/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_15 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_5/add (None, 197, 192) 384 ['dropout_15[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_4/add_
2[0][0]']
Transformer/EncoderBlock_5/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_5/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_5/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_5/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_5/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_5/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_16 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_5/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_5/Mlp (None, 197, 192) 148032 ['dropout_16[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_17 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_5/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_5/add (None, 197, 192) 384 ['Transformer/EncoderBlock_5/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_17[0][0]']
Transformer/EncoderBlock_6/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_5/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_6/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_6/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_6/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_18 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_6/add (None, 197, 192) 384 ['dropout_18[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_5/add_
2[0][0]']
Transformer/EncoderBlock_6/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_6/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_6/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_6/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_6/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_6/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_19 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_6/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_6/Mlp (None, 197, 192) 148032 ['dropout_19[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_20 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_6/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_6/add (None, 197, 192) 384 ['Transformer/EncoderBlock_6/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_20[0][0]']
Transformer/EncoderBlock_7/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_6/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_7/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_7/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_7/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_21 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_7/add (None, 197, 192) 384 ['dropout_21[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_6/add_
2[0][0]']
Transformer/EncoderBlock_7/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_7/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_7/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_7/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_7/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_7/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_22 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_7/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_7/Mlp (None, 197, 192) 148032 ['dropout_22[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_23 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_7/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_7/add (None, 197, 192) 384 ['Transformer/EncoderBlock_7/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_23[0][0]']
Transformer/EncoderBlock_8/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_7/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_8/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_8/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_8/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_24 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_8/add (None, 197, 192) 384 ['dropout_24[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_7/add_
2[0][0]']
Transformer/EncoderBlock_8/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_8/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_8/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_8/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_8/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_8/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_25 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_8/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_8/Mlp (None, 197, 192) 148032 ['dropout_25[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_26 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_8/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_8/add (None, 197, 192) 384 ['Transformer/EncoderBlock_8/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_26[0][0]']
Transformer/EncoderBlock_9/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_8/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_9/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_9/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_9/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_27 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_9/add (None, 197, 192) 384 ['dropout_27[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_8/add_
2[0][0]']
Transformer/EncoderBlock_9/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_9/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_9/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_9/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_9/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_9/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_28 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_9/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_9/Mlp (None, 197, 192) 148032 ['dropout_28[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_29 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_9/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_9/add (None, 197, 192) 384 ['Transformer/EncoderBlock_9/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_29[0][0]']
Transformer/EncoderBlock_10/La (None, 197, 192) 768 ['Transformer/EncoderBlock_9/add_
yerNorm_0 (QuantizedLayerNorma 2[0][0]']
lization)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/q erNorm_0[0][0]']
uery (QuantizedDense)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/k erNorm_0[0][0]']
ey (QuantizedDense)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/v erNorm_0[0][0]']
alue (QuantizedDense)
Transformer/EncoderBlock_10/Mu ((None, 197, 192), 384 ['Transformer/EncoderBlock_10/Mul
ltiHeadDotProductAttention_1/a (None, 3, 197, 197 tiHeadDotProductAttention_1/query
ttention (QuantizedAttention) )) [0][0]',
'Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/key[0
][0]',
'Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/value
[0][0]']
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_10/Mul
ltiHeadDotProductAttention_1/o tiHeadDotProductAttention_1/atten
ut (QuantizedDense) tion[0][0]']
dropout_30 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/out[0
][0]']
Transformer/EncoderBlock_10/ad (None, 197, 192) 384 ['dropout_30[0][0]',
d_1 (QuantizedAdd) 'Transformer/EncoderBlock_9/add_
2[0][0]']
Transformer/EncoderBlock_10/La (None, 197, 192) 768 ['Transformer/EncoderBlock_10/add
yerNorm_2 (QuantizedLayerNorma _1[0][0]']
lization)
Transformer/EncoderBlock_10/Ml (None, 197, 768) 148224 ['Transformer/EncoderBlock_10/Lay
pBlock/Dense_0 (QuantizedDense erNorm_2[0][0]']
)
Transformer/EncoderBlock_10/Ml (None, 197, 768) 1536 ['Transformer/EncoderBlock_10/Mlp
pBlock/activation (QuantizedRe Block/Dense_0[0][0]']
LU)
dropout_31 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_10/Mlp
Block/activation[0][0]']
Transformer/EncoderBlock_10/Ml (None, 197, 192) 148032 ['dropout_31[0][0]']
pBlock/Dense_1 (QuantizedDense
)
dropout_32 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_10/Mlp
Block/Dense_1[0][0]']
Transformer/EncoderBlock_10/ad (None, 197, 192) 384 ['Transformer/EncoderBlock_10/add
d_2 (QuantizedAdd) _1[0][0]',
'dropout_32[0][0]']
Transformer/EncoderBlock_11/La (None, 197, 192) 768 ['Transformer/EncoderBlock_10/add
yerNorm_0 (QuantizedLayerNorma _2[0][0]']
lization)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/q erNorm_0[0][0]']
uery (QuantizedDense)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/k erNorm_0[0][0]']
ey (QuantizedDense)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/v erNorm_0[0][0]']
alue (QuantizedDense)
Transformer/EncoderBlock_11/Mu ((None, 197, 192), 384 ['Transformer/EncoderBlock_11/Mul
ltiHeadDotProductAttention_1/a (None, 3, 197, 197 tiHeadDotProductAttention_1/query
ttention (QuantizedAttention) )) [0][0]',
'Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/key[0
][0]',
'Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/value
[0][0]']
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_11/Mul
ltiHeadDotProductAttention_1/o tiHeadDotProductAttention_1/atten
ut (QuantizedDense) tion[0][0]']
dropout_33 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/out[0
][0]']
Transformer/EncoderBlock_11/ad (None, 197, 192) 384 ['dropout_33[0][0]',
d_1 (QuantizedAdd) 'Transformer/EncoderBlock_10/add
_2[0][0]']
Transformer/EncoderBlock_11/La (None, 197, 192) 768 ['Transformer/EncoderBlock_11/add
yerNorm_2 (QuantizedLayerNorma _1[0][0]']
lization)
Transformer/EncoderBlock_11/Ml (None, 197, 768) 148224 ['Transformer/EncoderBlock_11/Lay
pBlock/Dense_0 (QuantizedDense erNorm_2[0][0]']
)
Transformer/EncoderBlock_11/Ml (None, 197, 768) 1536 ['Transformer/EncoderBlock_11/Mlp
pBlock/activation (QuantizedRe Block/Dense_0[0][0]']
LU)
dropout_34 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_11/Mlp
Block/activation[0][0]']
Transformer/EncoderBlock_11/Ml (None, 197, 192) 148032 ['dropout_34[0][0]']
pBlock/Dense_1 (QuantizedDense
)
dropout_35 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_11/Mlp
Block/Dense_1[0][0]']
Transformer/EncoderBlock_11/ad (None, 197, 192) 384 ['Transformer/EncoderBlock_11/add
d_2 (QuantizedAdd) _1[0][0]',
'dropout_35[0][0]']
Transformer/EncoderNorm (Quant (None, 197, 192) 1152 ['Transformer/EncoderBlock_11/add
izedBatchNormalization) _2[0][0]']
ExtractToken (QuantizedExtract (None, 192) 0 ['Transformer/EncoderNorm[0][0]']
Token)
Head (QuantizedDense) (None, 1000) 193000 ['ExtractToken[0][0]']
dequantizer (Dequantizer) (None, 1000) 0 ['Head[0][0]']
==================================================================================================
Total params: 5,773,912
Trainable params: 5,717,416
Non-trainable params: 56,496
__________________________________________________________________________________________________
The bc_vit_ti16_imagenet_pretrained helper was obtained with the same 8-bit quantization scheme but with an additional QAT step to further improve accuracy.
4.2 Quantization Aware Training (Optional)
In Section 4.1, we performed PTQ and converted the weights and activation outputs to 8-bit integer numbers. In most cases, there is no accuracy drop observed after quantization, however in cases where an accurary drop is observed, it is possible to further fine-tune this model using QAT.
The model that is obtained through QuantizeML python package is an instance of Keras. This allows the model to be fine-tuned using the original dataset to regain accuracy.
Akida models python package provides pre-trained models for vit_ti16 and deit_ti16 that have been trained using QAT method. It can be used in the following way:
from akida_models import bc_vit_ti16_imagenet_pretrained
# Load the pre-trained quantized model
model_quantized = bc_vit_ti16_imagenet_pretrained()
model_quantized.summary()
Downloading data from https://data.brainchip.com/models/AkidaV2/vit/bc_vit_ti16_224_i8_w8_a8.h5.
0/24413248 [..............................] - ETA: 0s
106496/24413248 [..............................] - ETA: 11s
704512/24413248 [..............................] - ETA: 3s
1392640/24413248 [>.............................] - ETA: 2s
2433024/24413248 [=>............................] - ETA: 1s
3784704/24413248 [===>..........................] - ETA: 1s
5144576/24413248 [=====>........................] - ETA: 1s
6840320/24413248 [=======>......................] - ETA: 0s
8749056/24413248 [=========>....................] - ETA: 0s
10846208/24413248 [============>.................] - ETA: 0s
13246464/24413248 [===============>..............] - ETA: 0s
15581184/24413248 [==================>...........] - ETA: 0s
18014208/24413248 [=====================>........] - ETA: 0s
20504576/24413248 [========================>.....] - ETA: 0s
22888448/24413248 [===========================>..] - ETA: 0s
24413248/24413248 [==============================] - 1s 0us/step
Download complete.
Model: "vit-tiny"
__________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
==================================================================================================
input (InputLayer) [(None, 224, 224, 3 0 []
)]
Rescale (QuantizedRescaling) (None, 224, 224, 3) 0 ['input[0][0]']
Embedding (QuantizedConv2D) (None, 14, 14, 192) 147648 ['Rescale[0][0]']
reshape (QuantizedReshape) (None, 196, 192) 0 ['Embedding[0][0]']
ClassToken (QuantizedClassToke (None, 197, 192) 192 ['reshape[0][0]']
n)
Transformer/PosEmbed (Quantize (None, 197, 192) 38208 ['ClassToken[0][0]']
dAddPositionEmbs)
Transformer/EncoderBlock_0/Lay (None, 197, 192) 768 ['Transformer/PosEmbed[0][0]']
erNorm_0 (QuantizedLayerNormal
ization)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_0/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_0/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_0/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_0/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_0/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_0/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_0/add (None, 197, 192) 384 ['dropout[0][0]',
_1 (QuantizedAdd) 'Transformer/PosEmbed[0][0]']
Transformer/EncoderBlock_0/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_0/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_0/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_0/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_0/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_0/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_1 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_0/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_0/Mlp (None, 197, 192) 148032 ['dropout_1[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_2 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_0/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_0/add (None, 197, 192) 384 ['Transformer/EncoderBlock_0/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_2[0][0]']
Transformer/EncoderBlock_1/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_0/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_1/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_1/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_1/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_1/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_1/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_3 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_1/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_1/add (None, 197, 192) 384 ['dropout_3[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_0/add_
2[0][0]']
Transformer/EncoderBlock_1/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_1/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_1/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_1/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_1/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_1/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_4 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_1/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_1/Mlp (None, 197, 192) 148032 ['dropout_4[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_5 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_1/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_1/add (None, 197, 192) 384 ['Transformer/EncoderBlock_1/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_5[0][0]']
Transformer/EncoderBlock_2/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_1/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_2/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_2/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_2/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_2/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_2/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_6 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_2/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_2/add (None, 197, 192) 384 ['dropout_6[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_1/add_
2[0][0]']
Transformer/EncoderBlock_2/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_2/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_2/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_2/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_2/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_2/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_7 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_2/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_2/Mlp (None, 197, 192) 148032 ['dropout_7[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_8 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_2/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_2/add (None, 197, 192) 384 ['Transformer/EncoderBlock_2/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_8[0][0]']
Transformer/EncoderBlock_3/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_2/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_3/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_3/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_3/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_3/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_3/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_9 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_3/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_3/add (None, 197, 192) 384 ['dropout_9[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_2/add_
2[0][0]']
Transformer/EncoderBlock_3/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_3/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_3/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_3/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_3/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_3/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_10 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_3/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_3/Mlp (None, 197, 192) 148032 ['dropout_10[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_11 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_3/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_3/add (None, 197, 192) 384 ['Transformer/EncoderBlock_3/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_11[0][0]']
Transformer/EncoderBlock_4/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_3/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_4/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_4/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_4/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_4/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_4/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_12 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_4/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_4/add (None, 197, 192) 384 ['dropout_12[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_3/add_
2[0][0]']
Transformer/EncoderBlock_4/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_4/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_4/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_4/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_4/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_4/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_13 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_4/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_4/Mlp (None, 197, 192) 148032 ['dropout_13[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_14 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_4/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_4/add (None, 197, 192) 384 ['Transformer/EncoderBlock_4/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_14[0][0]']
Transformer/EncoderBlock_5/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_4/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_5/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_5/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_5/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_5/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_5/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_15 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_5/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_5/add (None, 197, 192) 384 ['dropout_15[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_4/add_
2[0][0]']
Transformer/EncoderBlock_5/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_5/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_5/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_5/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_5/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_5/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_16 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_5/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_5/Mlp (None, 197, 192) 148032 ['dropout_16[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_17 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_5/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_5/add (None, 197, 192) 384 ['Transformer/EncoderBlock_5/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_17[0][0]']
Transformer/EncoderBlock_6/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_5/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_6/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_6/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_6/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_6/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_6/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_18 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_6/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_6/add (None, 197, 192) 384 ['dropout_18[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_5/add_
2[0][0]']
Transformer/EncoderBlock_6/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_6/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_6/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_6/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_6/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_6/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_19 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_6/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_6/Mlp (None, 197, 192) 148032 ['dropout_19[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_20 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_6/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_6/add (None, 197, 192) 384 ['Transformer/EncoderBlock_6/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_20[0][0]']
Transformer/EncoderBlock_7/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_6/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_7/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_7/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_7/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_7/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_7/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_21 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_7/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_7/add (None, 197, 192) 384 ['dropout_21[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_6/add_
2[0][0]']
Transformer/EncoderBlock_7/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_7/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_7/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_7/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_7/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_7/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_22 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_7/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_7/Mlp (None, 197, 192) 148032 ['dropout_22[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_23 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_7/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_7/add (None, 197, 192) 384 ['Transformer/EncoderBlock_7/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_23[0][0]']
Transformer/EncoderBlock_8/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_7/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_8/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_8/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_8/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_8/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_8/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_24 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_8/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_8/add (None, 197, 192) 384 ['dropout_24[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_7/add_
2[0][0]']
Transformer/EncoderBlock_8/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_8/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_8/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_8/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_8/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_8/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_25 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_8/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_8/Mlp (None, 197, 192) 148032 ['dropout_25[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_26 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_8/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_8/add (None, 197, 192) 384 ['Transformer/EncoderBlock_8/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_26[0][0]']
Transformer/EncoderBlock_9/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_8/add_
erNorm_0 (QuantizedLayerNormal 2[0][0]']
ization)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/qu rNorm_0[0][0]']
ery (QuantizedDense)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37058 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/ke rNorm_0[0][0]']
y (QuantizedDense)
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_9/Laye
tiHeadDotProductAttention_1/va rNorm_0[0][0]']
lue (QuantizedDense)
Transformer/EncoderBlock_9/Mul ((None, 197, 192), 384 ['Transformer/EncoderBlock_9/Mult
tiHeadDotProductAttention_1/at (None, 3, 197, 197 iHeadDotProductAttention_1/query[
tention (QuantizedAttention) )) 0][0]',
'Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/key[0]
[0]',
'Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/value[
0][0]']
Transformer/EncoderBlock_9/Mul (None, 197, 192) 37440 ['Transformer/EncoderBlock_9/Mult
tiHeadDotProductAttention_1/ou iHeadDotProductAttention_1/attent
t (QuantizedDense) ion[0][0]']
dropout_27 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_9/Mult
iHeadDotProductAttention_1/out[0]
[0]']
Transformer/EncoderBlock_9/add (None, 197, 192) 384 ['dropout_27[0][0]',
_1 (QuantizedAdd) 'Transformer/EncoderBlock_8/add_
2[0][0]']
Transformer/EncoderBlock_9/Lay (None, 197, 192) 768 ['Transformer/EncoderBlock_9/add_
erNorm_2 (QuantizedLayerNormal 1[0][0]']
ization)
Transformer/EncoderBlock_9/Mlp (None, 197, 768) 148224 ['Transformer/EncoderBlock_9/Laye
Block/Dense_0 (QuantizedDense) rNorm_2[0][0]']
Transformer/EncoderBlock_9/Mlp (None, 197, 768) 1536 ['Transformer/EncoderBlock_9/MlpB
Block/activation (QuantizedReL lock/Dense_0[0][0]']
U)
dropout_28 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_9/MlpB
lock/activation[0][0]']
Transformer/EncoderBlock_9/Mlp (None, 197, 192) 148032 ['dropout_28[0][0]']
Block/Dense_1 (QuantizedDense)
dropout_29 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_9/MlpB
lock/Dense_1[0][0]']
Transformer/EncoderBlock_9/add (None, 197, 192) 384 ['Transformer/EncoderBlock_9/add_
_2 (QuantizedAdd) 1[0][0]',
'dropout_29[0][0]']
Transformer/EncoderBlock_10/La (None, 197, 192) 768 ['Transformer/EncoderBlock_9/add_
yerNorm_0 (QuantizedLayerNorma 2[0][0]']
lization)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/q erNorm_0[0][0]']
uery (QuantizedDense)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/k erNorm_0[0][0]']
ey (QuantizedDense)
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_10/Lay
ltiHeadDotProductAttention_1/v erNorm_0[0][0]']
alue (QuantizedDense)
Transformer/EncoderBlock_10/Mu ((None, 197, 192), 384 ['Transformer/EncoderBlock_10/Mul
ltiHeadDotProductAttention_1/a (None, 3, 197, 197 tiHeadDotProductAttention_1/query
ttention (QuantizedAttention) )) [0][0]',
'Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/key[0
][0]',
'Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/value
[0][0]']
Transformer/EncoderBlock_10/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_10/Mul
ltiHeadDotProductAttention_1/o tiHeadDotProductAttention_1/atten
ut (QuantizedDense) tion[0][0]']
dropout_30 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_10/Mul
tiHeadDotProductAttention_1/out[0
][0]']
Transformer/EncoderBlock_10/ad (None, 197, 192) 384 ['dropout_30[0][0]',
d_1 (QuantizedAdd) 'Transformer/EncoderBlock_9/add_
2[0][0]']
Transformer/EncoderBlock_10/La (None, 197, 192) 768 ['Transformer/EncoderBlock_10/add
yerNorm_2 (QuantizedLayerNorma _1[0][0]']
lization)
Transformer/EncoderBlock_10/Ml (None, 197, 768) 148224 ['Transformer/EncoderBlock_10/Lay
pBlock/Dense_0 (QuantizedDense erNorm_2[0][0]']
)
Transformer/EncoderBlock_10/Ml (None, 197, 768) 1536 ['Transformer/EncoderBlock_10/Mlp
pBlock/activation (QuantizedRe Block/Dense_0[0][0]']
LU)
dropout_31 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_10/Mlp
Block/activation[0][0]']
Transformer/EncoderBlock_10/Ml (None, 197, 192) 148032 ['dropout_31[0][0]']
pBlock/Dense_1 (QuantizedDense
)
dropout_32 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_10/Mlp
Block/Dense_1[0][0]']
Transformer/EncoderBlock_10/ad (None, 197, 192) 384 ['Transformer/EncoderBlock_10/add
d_2 (QuantizedAdd) _1[0][0]',
'dropout_32[0][0]']
Transformer/EncoderBlock_11/La (None, 197, 192) 768 ['Transformer/EncoderBlock_10/add
yerNorm_0 (QuantizedLayerNorma _2[0][0]']
lization)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/q erNorm_0[0][0]']
uery (QuantizedDense)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37058 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/k erNorm_0[0][0]']
ey (QuantizedDense)
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_11/Lay
ltiHeadDotProductAttention_1/v erNorm_0[0][0]']
alue (QuantizedDense)
Transformer/EncoderBlock_11/Mu ((None, 197, 192), 384 ['Transformer/EncoderBlock_11/Mul
ltiHeadDotProductAttention_1/a (None, 3, 197, 197 tiHeadDotProductAttention_1/query
ttention (QuantizedAttention) )) [0][0]',
'Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/key[0
][0]',
'Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/value
[0][0]']
Transformer/EncoderBlock_11/Mu (None, 197, 192) 37440 ['Transformer/EncoderBlock_11/Mul
ltiHeadDotProductAttention_1/o tiHeadDotProductAttention_1/atten
ut (QuantizedDense) tion[0][0]']
dropout_33 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_11/Mul
tiHeadDotProductAttention_1/out[0
][0]']
Transformer/EncoderBlock_11/ad (None, 197, 192) 384 ['dropout_33[0][0]',
d_1 (QuantizedAdd) 'Transformer/EncoderBlock_10/add
_2[0][0]']
Transformer/EncoderBlock_11/La (None, 197, 192) 768 ['Transformer/EncoderBlock_11/add
yerNorm_2 (QuantizedLayerNorma _1[0][0]']
lization)
Transformer/EncoderBlock_11/Ml (None, 197, 768) 148224 ['Transformer/EncoderBlock_11/Lay
pBlock/Dense_0 (QuantizedDense erNorm_2[0][0]']
)
Transformer/EncoderBlock_11/Ml (None, 197, 768) 1536 ['Transformer/EncoderBlock_11/Mlp
pBlock/activation (QuantizedRe Block/Dense_0[0][0]']
LU)
dropout_34 (QuantizedDropout) (None, 197, 768) 0 ['Transformer/EncoderBlock_11/Mlp
Block/activation[0][0]']
Transformer/EncoderBlock_11/Ml (None, 197, 192) 148032 ['dropout_34[0][0]']
pBlock/Dense_1 (QuantizedDense
)
dropout_35 (QuantizedDropout) (None, 197, 192) 0 ['Transformer/EncoderBlock_11/Mlp
Block/Dense_1[0][0]']
Transformer/EncoderBlock_11/ad (None, 197, 192) 384 ['Transformer/EncoderBlock_11/add
d_2 (QuantizedAdd) _1[0][0]',
'dropout_35[0][0]']
Transformer/EncoderNorm (Quant (None, 197, 192) 1152 ['Transformer/EncoderBlock_11/add
izedBatchNormalization) _2[0][0]']
ExtractToken (QuantizedExtract (None, 192) 0 ['Transformer/EncoderNorm[0][0]']
Token)
Head (QuantizedDense) (None, 1000) 193000 ['ExtractToken[0][0]']
dequantizer (Dequantizer) (None, 1000) 0 ['Head[0][0]']
==================================================================================================
Total params: 5,773,912
Trainable params: 5,717,416
Non-trainable params: 56,496
__________________________________________________________________________________________________
5. Conversion to Akida
A model quantized through QuantizeML python package is ready to be converted to Akida. Once the quantized model has the desired accuracy CNN2SNN toolkit is used for conversion to Akida. There is no further optimization required and equivalent accuracy is observed upon converting the model to Akida.
from cnn2snn import convert
# Convert the model
model_akida = convert(model_quantized)
model_akida.summary()
Model Summary
________________________________________________
Input shape Output shape Sequences Layers
================================================
[224, 224, 3] [1, 1, 1000] 1 14
________________________________________________
_______________________________________________________________________
Layer (type) Output shape Kernel shape
================= SW/Embedding-dequantizer (Software) =================
Embedding (Stem) [1, 197, 192] (16, 16, 3, 192)
_______________________________________________________________________
VitEncoderBlock_2 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_3 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_4 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_5 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_6 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_7 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_8 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_9 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_10 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_11 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_12 (VitEncoderBlock) [1, 197, 192] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
_______________________________________________________________________
VitEncoderBlock_13 (VitEncoderBlock) [1, 1, 1000] N/A
norm_mha N/A
query (192, 192)
key (192, 192)
value (192, 192)
attention N/A
attention_projection (192, 192)
skip_connection_1 N/A
norm_mlp N/A
mlp_1 (192, 768)
mlp_2 (768, 192)
skip_connection_2 N/A
batch_norm N/A
extract_token N/A
head (192, 1000)
_______________________________________________________________________
dequantizer (Dequantizer) [1, 1, 1000] N/A
_______________________________________________________________________
6. Displaying results Attention Maps
Instead of showing predictions, here we propose to show attention maps on an image. This is derived from Abnar et al. attention rollout as shown in the following Keras tutorial. This aims to highlight the model abilities to focus on relevant parts in the input image.
Just like for the AkidaNet example, ImageNet images are not publicly available, this example uses a set of 10 copyright free images that were found on Google using ImageNet class names.
Get the preprocessed sample images:
import numpy as np
from akida_models.imagenet import get_preprocessed_samples
# Model specification and hyperparameters
NUM_CHANNELS = 3
IMAGE_SIZE = 224
NUM_IMAGES = 10
# Load the preprocessed images
x_test, _ = get_preprocessed_samples(IMAGE_SIZE, NUM_CHANNELS)
print(f'{NUM_IMAGES} images loaded and preprocessed.')
10 images loaded and preprocessed.
Build and display the attention map for one selected sample:
import cv2
import matplotlib.pyplot as plt
from keras import Model
from quantizeml.layers import ClassToken, Attention
from quantizeml.tensors import FixedPoint
from quantizeml.models.transforms.transforms_utils import get_layers_by_type
def build_attention_map(model, image):
# Get the Attention layers list
attentions = get_layers_by_type(model, Attention)
# Calculate the number of tokens and deduce the grid size
num_tokens = sum(isinstance(ly, ClassToken) for ly in model.layers)
grid_size = int(np.sqrt(attentions[0].output_shape[0][-2] - num_tokens))
# Get the attention weights from each transformer
outputs = [la.output[1] for la in attentions]
weights = Model(inputs=model.inputs, outputs=outputs).predict(np.expand_dims(image, 0))
# Converts to float if needed
weights = [w.to_float() if isinstance(w, FixedPoint) else w for w in weights]
weights = np.array(weights)
# Heads number
num_heads = weights.shape[2]
num_layers = weights.shape[0]
reshaped = weights.reshape((num_layers, num_heads, grid_size**2 + 1, grid_size**2 + 1))
# Average the attention weights across all heads
reshaped = reshaped.mean(axis=1)
# To account for residual connections, we add an identity matrix to the attention matrix and
# re-normalize the weights.
reshaped = reshaped + np.eye(reshaped.shape[1])
reshaped = reshaped / reshaped.sum(axis=(1, 2))[:, np.newaxis, np.newaxis]
# Recursively multiply the weight matrices
v = reshaped[-1]
for n in range(1, len(reshaped)):
v = np.matmul(v, reshaped[-1 - n])
# Attention from the output token to the input space
mask = v[0, 1:].reshape(grid_size, grid_size)
mask = cv2.resize(mask / mask.max(), (image.shape[1], image.shape[0]))[..., np.newaxis]
return (mask * image).astype("uint8")
# Using a specific image for which attention map is easier to observe
image = x_test[8]
# Compute the attention map
attention_float = build_attention_map(model_keras, image)
attention_quantized = build_attention_map(model_quantized, image)
# Display the attention map
fig, (ax1, ax2, ax3) = plt.subplots(ncols=3)
ax1.axis('off')
ax1.set_title('Original')
ax1.imshow(image)
ax2.axis('off')
ax2.set_title('Float')
ax2.imshow(attention_float)
ax3.axis('off')
ax3.set_title('Quantized')
ax3.imshow(attention_quantized)
fig.suptitle('Attention masks', fontsize=10)
plt.show()
1/1 [==============================] - ETA: 0s
1/1 [==============================] - 2s 2s/step
1/1 [==============================] - ETA: 0s
1/1 [==============================] - 37s 37s/step
Total running time of the script: (3 minutes 14.384 seconds)