Added serial warning suppression to TrainException because we don't care
(Like in assignment 1).