Getting an error AttributeError: module 'tensorflow' has no attribute 'data' " when I try to run model retraining code