Commit 6efa4c3d authored by Ekanayake G.B.E.M.K's avatar Ekanayake G.B.E.M.K

Xception model

parent 15308e07
...@@ -172,6 +172,7 @@ ...@@ -172,6 +172,7 @@
" x = Dropout(rate)(x)\n", " x = Dropout(rate)(x)\n",
" return x\n", " return x\n",
"\n", "\n",
" #define xception model\n",
" def model_conversion(self, trainable):\n", " def model_conversion(self, trainable):\n",
" functional_model = tf.keras.applications.Xception(weights=\"imagenet\")\n", " functional_model = tf.keras.applications.Xception(weights=\"imagenet\")\n",
" functional_model.trainable = trainable\n", " functional_model.trainable = trainable\n",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment