--runtime = 43.212
--epoch_runtime = 0.847