onEpochEnd

open override fun onEpochEnd(epoch: Int, context: ComputationContext)