Added training for LRGB