add a workaround against that invalid nnabla_pb2.Variable can come (#828

) - originally, NnpExpander should be fixed
sony · Apr 21, 2018 · d63b8a2 · d63b8a2
1 parent 2a658a2
commit d63b8a2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/python/src/nnabla/utils/converter/nnablart/nnb.py b/python/src/nnabla/utils/converter/nnablart/nnb.py
@@ -105,7 +105,15 @@ def export(self, *args):
                     index, pointer = self._alloc(data=data)
                     var.data_index = index
                 elif v.type == 'Buffer':
-                    var.data_index = (vidx_to_abidx[n] + 1) * -1
+                    # FIXME: remove the following workaround
+                    if n in vidx_to_abidx:
+                        # n which is NOT in vidx_to_abidx can appear
+                        # since NnpExpander doesn't handle --nnp-expand-network correctly
+                        var.data_index = (vidx_to_abidx[n] + 1) * -1
+                    else:
+                        # this var doesn't make sense, but add  it
+                        # so that nn_network_t::variables::size is conserved
+                        var.data_index = -1
 
                 variable = struct.pack('IiIBi',
                                        var.id,