--runtime = 301.182
--epoch_runtime = 2.982