Error loading '.net' model with netx.hdf5

[ZoltanV-V]

Hello, I recently updated to the latest version of lava-dl and lava and I am coming across this error when trying to load my '.net' model with netx:

File "lava-dl/tutorials/lava/lib/dl/bootstrap/mnist/mnist.py", line 386, in
net = hdf5.Network(net_config=trained_folder + '/network.net')
File "./lava-dl-env/lib/python3.8/site-packages/lava/magma/core/process/process.py", line 33, in call
obj = type.call(cls, *args, **kwargs)
File "./lava-dl-env/lib/python3.8/site-packages/lava/lib/dl/netx/hdf5.py", line 76, in init
self.layers = self._create()
File "./lava-dl-env/lib/python3.8/site-packages/lava/lib/dl/netx/hdf5.py", line 444, in _create
layer, table = self.create_input(
File "./lava-dl-env/lib/python3.8/site-packages/lava/lib/dl/netx/hdf5.py", line 263, in create_input
height=shape[1],
IndexError: tuple index out of range

I am just running the mnist tutorial for bootstrap so exporting the model with this line:
net.export_hdf5(trained_folder + '/network.net')

And following the netx tutorial, I'm loading it with this line:
net = hdf5.Network(net_config=trained_folder + '/network.net')

My h5py version is 3.7.0

Any suggestion on how to resolve the issue would be appreciated. Thank you!

Objective of issue:

Lava DL version:

• 0.3.2
• 0.3.0 (feature release)
• 0.2.1 (bug fixes)
• 0.2.0 (current version)
• 0.1.2

Lava version:

• 0.5
• 0.4.0 (feature release)
• 0.3.1 (bug fixes)
• 0.3.0 (current version)
• 0.2.0
• 0.1.2

I'm submitting a ...

• bug report
• feature request
• documentation request

Current behavior:

Expected behavior:

Steps to reproduce:

Related code:

insert short code snippets here

Other information:

insert the output from lava debug here

[bamsumit]

@ZoltanV-V thanks for reporting it. We will look into the issue.

[bamsumit]

@ZoltanV-V We have fixed the issue. Can you try the latest code and see if it works for you?

Since you are already looking at the MNIST inference, it would be great if you can contribute an inference notebook for it.

[ZoltanV-V]

@bamsumit I've just tried it with the update and it works now, thank you! I would be happy to contribute an inference notebook. Thanks again!

[bamsumit]

Great! Thanks for the update.