--runtime = 648.107
--epoch_runtime = 58.919