weights not saved properly
sushant7878 opened this issue · 1 comments
sushant7878 commented
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 commented
I got the solution. Just install tensorflow-gpu==2.3.1 instead of tensorflow==2.3.0rc0
and
addition install
tensorflow==2.3.1