MGEloss class aixd.mlmodel.architecture.losses.MGEloss(*args: Any, **kwargs: Any)[source] Bases: Module Mean gradient error. Uses a sobel filter to compute the gradient of predicted and estimated images, and then applies a mean square difference of the pixel-wise gradients. Methods forward