ImportError: `save_weights` requires h5py when saving in hdf5

When running training the demo image, after epoch 1 is completed, the following error appeared.

ImportError: save_weights requires h5py when saving in hdf5.

However, h5py is already installed with pip.

I fixed it by
pip uninstall h5py
conda install h5py -c conda-forge

reference:
tensorflow/tensorflow#60464

Thanks

[jdeschamps]

Hi,

Sorry for the late reply, we will have a look!