**Describe the bug** test.py return "W tensorflow/core/framework/op_kernel.cc:1192] Not found: Unsuccessful TensorSliceReader constructor: ". **To Reproduce** Steps to reproduce the behavior: 1. Run python3 watermarks.py --logdir=save/ 2. Run python3 tests.py **Expected behavior** Test pass.