weights not saved properly

[sushant7878]

when I try to run save_model.py it runs without error but all the weights are not saved properly. it says

Total params: 6,062,814 Trainable params: 6,056,606 Non-trainable params: 6,208

when it is saved properly, it should be

Total params: 64,429,405 Trainable params: 64,363,101 Non-trainable params: 66,304

as shown in the figure below.

can someone help me with the case??

[sushant7878]

I got the solution. Just install tensorflow-gpu==2.3.1 instead of tensorflow==2.3.0rc0

and

addition install

tensorflow==2.3.1