fix the retrain model to eager closes #78
This commit is contained in:
@@ -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]
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user