TensorFlow： mode.save()报错 non-trackable object: (None, None)

问题描述

环境：tensorflow2.2.0

执行model.save(), 报错  ...non-trackable object: (None, None)...

解决办法

搞了半天没有找到好的办法，只能通过升级搞定

pip uninstall tensorflow
pip install tensorflow==2.6.0
pip install --upgrade keras==2.6.0

然后就搞定了

问题描述

使用pickle保存model的fit后的history，报错：cannot pickle 'weakref' object

解决办法

报错代码

with open(history_path, 'wb+') as h_obj:
    pickle.dump(history, h_obj)

修改为

with open(history_path, 'wb+') as h_obj:
    pickle.dump(history.history, h_obj)