--runtime = 1625.14
--epoch_runtime = 16.09