Pass logits and labels to losses using kwargs
All threads resolved!
All threads resolved!
Compare changes
- Amir MOHAMMADI authored
@@ -80,12 +80,12 @@ class Logits(estimator.Estimator):
@@ -144,7 +144,7 @@ class Logits(estimator.Estimator):
@@ -187,7 +187,7 @@ class Logits(estimator.Estimator):
@@ -266,7 +266,7 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -275,13 +275,13 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -365,7 +365,7 @@ class LogitsCenterLoss(estimator.Estimator):