update private loss calculation
