Refactor loss arguments.

rayanirban · Apr 1, 2021 · 2515356 · 2515356
1 parent bb9ea43
commit 2515356
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fit/modules/loss.py b/fit/modules/loss.py
@@ -3,9 +3,9 @@
 from fit.utils.utils import denormalize_amp, denormalize_phi
 
 
-def _fc_prod_loss(pred_fc, target_fc, mag_min, mag_max):
-    pred_amp = denormalize_amp(pred_fc[..., 0], amp_min=mag_min, amp_max=mag_max)
-    target_amp = denormalize_amp(target_fc[..., 0], amp_min=mag_min, amp_max=mag_max)
+def _fc_prod_loss(pred_fc, target_fc, amp_min, amp_max):
+    pred_amp = denormalize_amp(pred_fc[..., 0], amp_min=amp_min, amp_max=amp_max)
+    target_amp = denormalize_amp(target_fc[..., 0], amp_min=amp_min, amp_max=amp_max)
 
     pred_phi = denormalize_phi(pred_fc[..., 1])
     target_phi = denormalize_phi(target_fc[..., 1])