Add COMET to the evaluation steps

[gregtatum]

We already use COMET in models repo, but don't have it in the training steps. This makes it hard to cross-compare with our release criteria while training models.