Estimator that loads variables from another model
Compare changes
- Tiago de Freitas Pereira authored
@@ -16,8 +16,8 @@ from bob.learn.tensorflow.network.utils import append_logits
@@ -43,6 +43,15 @@ class Logits(estimator.Estimator):
@@ -70,6 +79,9 @@ class Logits(estimator.Estimator):
@@ -81,6 +93,8 @@ class Logits(estimator.Estimator):
@@ -89,6 +103,7 @@ class Logits(estimator.Estimator):
@@ -107,18 +122,36 @@ class Logits(estimator.Estimator):
@@ -131,16 +164,7 @@ class Logits(estimator.Estimator):
@@ -156,6 +180,7 @@ class Logits(estimator.Estimator):
@@ -200,6 +225,10 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -212,6 +241,8 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -221,6 +252,7 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -237,17 +269,42 @@ class LogitsCenterLoss(estimator.Estimator):
@@ -255,27 +312,11 @@ class LogitsCenterLoss(estimator.Estimator):