--runtime = 3931.34
--epoch_runtime = 78.627