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

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

Open CharlesGaydon opened this issue 11 months ago • 0 comments

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

https://github.com/IGNF/myria3d/blob/8c01bdf0424f19b1cc8ef4985a6735d15b59653e/myria3d/models/model.py#L74

Mar 20 '24 12:03 CharlesGaydon