Made loss augmentation in HardEM a flag.