init: Return status of restore

Author: revsic

model/__init__.py

@@ -121,4 +121,4 @@ def restore(self, path, optim=None):
         if optim is not None:
             kwargs['optim'] = optim
         ckpt = tf.train.Checkpoint(**kwargs)
-        ckpt.restore(path)
+        return ckpt.restore(path)