I\'ve been tasked with creating a training model which will produce and .h5 file at the end. I\'ve been having a lot of trouble with this however and was hoping I could get some