pass gamma through for training