Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Problems loading the model #9

Open
steve2ag opened this issue Oct 4, 2019 · 0 comments
Open

Problems loading the model #9

steve2ag opened this issue Oct 4, 2019 · 0 comments

Comments

@steve2ag
Copy link

steve2ag commented Oct 4, 2019

I´ve been trying to load the model with the weigths. I use the following commands

with tf.Session() as sess:
#Obtaining the graph
saver = tf.train.import_meta_graph('checkpoint/1_96/ds_ft_hybrid_4ct.model-34000.meta')
#Restoring the weights
saver.restore(sess,'checkpoint/1_96')

The problem arises while restoring the weigths. I keep getting the following error

InvalidArgumentError: Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Cannot assign a device for operation deconv1_1/deconv/filter/Initializer/random_normal/RandomStandardNormal: node deconv1_1/deconv/filter/Initializer/random_normal/RandomStandardNormal (defined at :3) was explicitly assigned to /device:GPU:1 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0 ]. Make sure the device specification refers to a valid device.
[[deconv1_1/deconv/filter/Initializer/random_normal/RandomStandardNormal]]

It would bee a lot of help if sameone could tell me if I´m doing something wrong
Thanks beforehand for your help, and congratulations for your grat work.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant