Strange error when loading model from tensorflow

kman88

Hello, I am getting an error when try to load a model using tensorflow. The error is this:

Key UNet/conv2_up/batch_normalization/gamma/Adam not found in checkpoint
     [[node save/RestoreV2_154 (defined at /tmp/spark-e3d4ddb7-ea57-4a68-abe0-72d64992c3a9/userFiles-9cdaee1d-d0b7-4a56-b509-4f9c9f84b07f/rec-PXpIbIJZ.zip/deploy_network_ao_modified.py:32)

The line that gives me the error is this:

saver.restore(sess, '{0}'.format(model_path))

The model does contain Adam, and similarly I'm able to run this to completion on my laptop, which has an identical version of tensorflow. Any ideas what's going on and how to remedy this error? I have read on help forums it might have something to do with old checkpoints but I'm not sure. Any help would be great.

kman88

Interestingly enough I am able to run this line in the notebook, just not as a job.