It doesn’t look like there’s anything wrong with your code. Looking at the TensorFlow docs, this is how TensorFlow saves models: “The variables directory contains a standard training checkpoint… The assets directory contains files used by the TensorFlow graph…” If you really want a single file for the saved model, you can use HDF5 format.
3 Likes