added option to use the amplitude in the loss: great improvements !!