hello,i have a question on the code in dl_trainer.py

[wheeOuO]

i want to use MG-WFBP to train the ResNet-50 model,but when I ran the code in Linux, an error occurred, which prompted me ‘’ModuleNotFoundError: No module named 'DatasetHDF5' ‘’.I see "from datasets import DatasetHDF5" in Line27 in dl_trainer.py,and i tried to use "pip install" and "conda install" to install this module,but it failed.How to solve this problem?

[shyhuai]

Hi, it seems that this file is incorrectly ignored by .gitignore. I have uploaded datasets.py. Please pull and try again. Thanks.

[wheeOuO]

Thanks! But there's a new error"ValueError: The SWMR feature is not available in this version of the HDF5 library".

[shyhuai]

Please set swmr=False if you are using a new version of hdf5 at: https://github.com/HKBU-HPML/MG-WFBP/blob/master/datasets.py#L14.