--runtime = 2220.753
--epoch_runtime = 25.236