eaplatanios / tensorflow_scala

TensorFlow API for the Scala Programming Language
http://platanios.org/tensorflow_scala/
Apache License 2.0
937 stars 95 forks source link

Error during tf.Saver.fromMetaGraphDef(metaGraphDef), KindCase #33

Closed lucataglia closed 6 years ago

lucataglia commented 6 years ago

Here is my error: 1

I'm trying to reload a previous trained TF Model with the Scala API. I checked that inside my my-model.meta file the variables collections is store inside a byte list. 2

Looking at the source code I found that here is where the kind of the collection is retrieve 3

And here I found out the kind associated to byte list is the number 2 4

My question is, the if (kind != 1) correct ? I expected if (kind != 2) than raise error.

eaplatanios commented 6 years ago

@lucaRadicalbit You are right. Good catch! :) This should have been fixed now.