param tweak
Update for new losses