no storing of the datasets anymore; method for finding the best experiment; only store one checkpoint; no casting to array from hdf5

True, I just left it in for the different outputs. But of course, one could also have that output and just one line calling trainer.fit().