Update RNTNTrainer.scala