--runtime = 3245.35
--epoch_runtime = 40.066