Only return / plot the cost function intermittently or when requested

[carterbox]

Computing the full cost function requires communications and reductions overhead which may not be needed.