Add loss weight for noise