--runtime = 1241.318
--epoch_runtime = 112.847