Estimators optimize loss
Compare changes
- Amir MOHAMMADI authored
+ 289
− 177
@@ -6,13 +6,17 @@ import tensorflow as tf
@@ -88,19 +92,28 @@ class Logits(estimator.Estimator):
@@ -109,128 +122,191 @@ class Logits(estimator.Estimator):
@@ -238,13 +314,12 @@ class Logits(estimator.Estimator):
@@ -260,19 +335,23 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -282,107 +361,135 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -391,42 +498,47 @@ class LogitsCenterLoss(estimator.Estimator):