Training does not produce good results

[Matthew Broadhead]

Here is some h5 data I have prepared
https://drive.google.com/file/d/17C7c6Sf6odq090vldRZPtk9odwlqVb6T/view?usp=sharing

train.py seems to run well but during test.py the results are not good. Even on the original models. Can someone help me understand if there is a problem with the data?