diff --git a/tests/test_losses.py b/tests/test_losses.py index d456261..78e7a2c 100644 --- a/tests/test_losses.py +++ b/tests/test_losses.py @@ -1,4 +1,5 @@ """ Test loss functions""" + from dgmr.losses import SSIMLoss, MS_SSIMLoss, SSIMLossDynamic, tv_loss import torch