fix the retrain model to eager closes #78

This commit is contained in:
2024-04-14 16:51:15 +01:00
parent 5f11126acd
commit 095a37189d
2 changed files with 3 additions and 3 deletions

View File

@@ -49,7 +49,7 @@ def pathToLabel(path):
num = table.lookup(tf.strings.as_string([path]))
return tf.cond(
tf.math.equal(num, tf.constant(-1)),
tf.math.equal(num, tf.constant(-2)),
lambda: tf.zeros([depth]),
lambda: tf.one_hot(table.lookup(tf.strings.as_string([path])) - diff, depth)[0]
)