Resolve "train.h5 file too big, needs to replace with smaller version"

Nazish Qamarrequested to merge
113-train-h5-file-too-big-needs-to-replace-with-smaller-version into master

Closes #113 (closed)

Merge request reports