I have been getting this error : UnimplementedError: Cast string to float is not supported [[node categorical_crossentropy/Cast (defined at :2) ]] [Op:__inference_train_fun