aimodelshare 0.1.35__py3-none-any.whl → 0.1.37__py3-none-any.whl

aimodelshare/aimsonnx.py

@@ -657,23 +657,41 @@ def _keras_to_onnx(model, transfer_learning=None,
             pass  # fallback, don't crash conversion
             
     keras_layers = keras_unpack(model)
+    
+    
+    from tensorflow.python.framework import tensor_shape  # <- place this at the top of your file
+    
     layers = []
     layers_n_params = []
     layers_shapes = []
     activations = []
-
+    
     for layer in keras_layers:
+        # layer name
         layers.append(layer.__class__.__name__)
-        layers_n_params.append(layer.count_params())
     
+        # parameter count
+        try:
+            layers_n_params.append(layer.count_params())
+        except:
+            layers_n_params.append(0)
+    
+        # output shape (sanitized for JSON)
         shape = getattr(layer, 'output_shape', None)
-        if shape is not None:
+    
+        if isinstance(shape, tensor_shape.TensorShape):
+            shape = shape.as_list()
+        elif shape is not None:
             try:
-                shape = tuple(shape)
+                shape = list(shape)
             except:
                 shape = str(shape)
+        else:
+            shape = None
+    
         layers_shapes.append(shape)
     
+        # activation
         if hasattr(layer, 'activation'):
             act = getattr(layer.activation, '__name__', None)
             if act:

aimodelshare/data_sharing/download_data.py

@@ -241,7 +241,7 @@ def import_quickstart_data(tutorial, section="modelplayground"):
         existing_folder = 'dog_competition_data'
 
     if all([tutorial == "imdb", section == "modelplayground"]):
-        quickstart_repository = "public.ecr.aws/y2e2a1d6/imdb_quickstart_materials-repository:latest"   
+        quickstart_repository = "public.ecr.aws/z5w0c9e9/imdb_quickstart_materials-repository:latest"
         existing_folder = 'imdb_competition_data'
 
     download_data(quickstart_repository)

{aimodelshare-0.1.35.dist-info → aimodelshare-0.1.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aimodelshare
-Version: 0.1.35
+Version: 0.1.37
 Summary: Deploy locally saved machine learning models to a live rest API and web-dashboard.  Share it with the world via modelshare.org
 Home-page: https://www.modelshare.org
 Author: Michael Parrott

{aimodelshare-0.1.35.dist-info → aimodelshare-0.1.37.dist-info}/RECORD

@@ -1,6 +1,6 @@
 aimodelshare/README.md,sha256=_OMdUIeIYZnpFlKdafM1KNWaANO2nWdx0QpLE_ZC-Qs,2014
 aimodelshare/__init__.py,sha256=CS0iFxgAic21gBcQE6NSZ-D_ElHw80_A3OWrEYo9Dks,539
-aimodelshare/aimsonnx.py,sha256=LW5uiIqF-nLIf0IivRaVa_b_PSBMmgeXF0lyPbEQCW0,67715
+aimodelshare/aimsonnx.py,sha256=oFPRaZ6HGqj8m9l1Zq1cBPMk9cH3QGy8Ru1c8xsQ9rY,68162
 aimodelshare/api.py,sha256=jeCIMbpBllTYi-bPvCdNvI6yHaT3JYakq9fgaz10s_E,34920
 aimodelshare/aws.py,sha256=jn99R9-N77Qac-_eYm-LaCQUPd-RnE7oVULm9rh-3RY,15232
 aimodelshare/aws_client.py,sha256=Ce19iwf69BwpuyyJlVN8z1da3c5jf93svsTgx1OWhaA,6784
@@ -31,7 +31,7 @@ aimodelshare/containerization_templates/lambda_function.txt,sha256=nEFoPDXemNcQZ
 aimodelshare/custom_approach/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 aimodelshare/custom_approach/lambda_function.py,sha256=d1HZlgviHZq4mNBKx4q-RCunDK8P8i9DKZcfv6Nmgzc,479
 aimodelshare/data_sharing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aimodelshare/data_sharing/download_data.py,sha256=SshJCz3jJaL1gqZmWjFMOKzUVhijqsisnPGA6jYXbq8,22795
+aimodelshare/data_sharing/download_data.py,sha256=6npwGwTOGYrNEP1CblS9RupzFMR0ATXxqZ24poLlmvY,22792
 aimodelshare/data_sharing/share_data.py,sha256=dMOP0-PTSpviOeHi3Nvj-uiq5PlIfk_SN5nN92j4PnI,13964
 aimodelshare/data_sharing/utils.py,sha256=865lN8-oGFi_U_zRaNnGB8Bd0sC8dN_iI5krZOSt_Ts,236
 aimodelshare/data_sharing/data_sharing_templates/Dockerfile.txt,sha256=27wmp7b0rXqJQsumhPxCvGHmUcDiiVgrC6i7DmY7KQA,77
@@ -140,11 +140,11 @@ aimodelshare/sam/codepipeline_policies.txt,sha256=267HMXMnbP7qRASkmFZYSx-2HmKf5o
 aimodelshare/sam/codepipeline_trust_relationship.txt,sha256=yfPYvZlN3fnaIHs7I3ENMMveigIE89mufV9pvR8EQH8,245
 aimodelshare/sam/spark-class.txt,sha256=chyJBxDzCzlUKXzVQYTzuJ2PXCTwg8_gd1yfnI-xbRw,217
 aimodelshare/sam/template.txt,sha256=JKSvEOZNaaLalHSx7r9psJg_6LLCb0XLAYi1-jYPu3M,1195
-aimodelshare-0.1.35.dist-info/licenses/LICENSE,sha256=JXBYLriXYgTloZs-9CJPZY76dqkuDT5df_HghMnljx8,1134
+aimodelshare-0.1.37.dist-info/licenses/LICENSE,sha256=JXBYLriXYgTloZs-9CJPZY76dqkuDT5df_HghMnljx8,1134
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_aimsonnx.py,sha256=-GOF1_qXGQaMxHyqK0GPg7dD1meE-S7CZea4pLmBDTk,3906
 tests/test_playground.py,sha256=vdFWPRrZNQ2poiBOoN3l7HsXB5yc3p3rrrclNYJHnaw,24574
-aimodelshare-0.1.35.dist-info/METADATA,sha256=zNTXlZRp8nRZcB635yerfl2VBgu2hnwZXSLGAQG54Gs,2916
-aimodelshare-0.1.35.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-aimodelshare-0.1.35.dist-info/top_level.txt,sha256=2KJgeHQ0BmZuilB75J203i7W4vri6CON2kdbwk9BNpU,19
-aimodelshare-0.1.35.dist-info/RECORD,,
+aimodelshare-0.1.37.dist-info/METADATA,sha256=TbdF7DWdlJGUwmOIw7RtsmP9Gn7N7C0ZKWHdHs4Yt3c,2916
+aimodelshare-0.1.37.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+aimodelshare-0.1.37.dist-info/top_level.txt,sha256=2KJgeHQ0BmZuilB75J203i7W4vri6CON2kdbwk9BNpU,19
+aimodelshare-0.1.37.dist-info/RECORD,,