SHARE
TWEET

Untitled

a guest Jul 24th, 2019 59 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. MODEL_DIR = "/Users/amit/Desktop/test/exportpath"
  2. version = 1
  3. export_path = os.path.join(MODEL_DIR, str(version))
  4. print('export_path = {}\n'.format(export_path))
  5. if os.path.isdir(export_path):
  6.   print('\nAlready saved a model, cleaning up\n')
  7. with tf.Session(graph=graph) as sess:
  8.     tf.saved_model.simple_save(
  9.         sess,
  10.         export_path,
  11.         inputs={'input_image': x},
  12.         outputs={'final_output':y})
  13.  
  14. print('\nSaved model:')
RAW Paste Data
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy. OK, I Understand
 
Top