changed GlobalAveragePooling2D to AveragePooling2D

llvm/projects/keras/src/mobilenet_cifar10.py

-
 import sys
 import os
 os.environ['CUDA_VISIBLE_DEVICES'] = '1'
@@ -102,7 +101,8 @@ def get_mobilenet(alpha=1, depth_multiplier=1):
     _depthwise_conv_block(1024, alpha, depth_multiplier)
     model.add(Dropout(rate=0.5))
 
-    model.add(GlobalAveragePooling2D())
+    model.add(AveragePooling2D(pool_size=2))
+    model.add(Flatten())
     model.add(Dense(10, activation='softmax'))
 
     return model