From 1aca5f36f556f8037000b60a9b0db0879b38f2cf Mon Sep 17 00:00:00 2001 From: hsharif3 <hsharif3@illinois.edu> Date: Tue, 16 Mar 2021 07:34:12 +0000 Subject: [PATCH] Update Support.md in Keras --- hpvm/projects/keras/docs/Support.md | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/hpvm/projects/keras/docs/Support.md b/hpvm/projects/keras/docs/Support.md index a31d012d0b..e5e7b1a1a2 100644 --- a/hpvm/projects/keras/docs/Support.md +++ b/hpvm/projects/keras/docs/Support.md @@ -1,5 +1,4 @@ - ## Supported Keras Operators The Keras frontend supports `Sequential()` Keras models. @@ -23,7 +22,19 @@ The list of supported operations is as follows: ## Limitations -* We support convolutional neural networks that include the supported operators above - RNNs/LSTMS are not supported +* Currently, we support Convolutional Neural Networks (CNNs) that include the supported operators (above) - RNNs/LSTMs not supported * We currently only support models in NCHW format (NHWC is not supported) * Softmax operator should be the last operation in the CNN pipeline -* Softmax operation must be a separate operator (not specified as activation to another type of Keras operator) +* Softmax operation must be a separate operator (not specified as activation to another type of Keras operator). Example of what works: + +``` +Activation ("softmax") +``` + +Example of what is NOT supported: + +``` +Dense(num_classes, activation="softmax") +``` + + -- GitLab