Move instantiation of metrics / CELoss to init method to avoid numerous moves to model device

[CharlesGaydon]

Use less .to(self.device) by instantiating everything in the init function.

myria3d/myria3d/models/model.py

Line 74 in 8c01bdf

self.train_iou = MulticlassJaccardIndex(self.hparams.num_classes).to(self.device)