Myria3D: Aerial Lidar HD Semantic Segmentation with Deep Learning
BSD 3-Clause "New" or "Revised" License
151
stars
20
forks
source link
Move instantiation of metrics / CELoss to init method to avoid numerous moves to model device #115
Open
CharlesGaydon opened 3 months ago
Use less
.to(self.device)by instantiating everything in the init function.https://github.com/IGNF/myria3d/blob/8c01bdf0424f19b1cc8ef4985a6735d15b59653e/myria3d/models/model.py#L74