Closed wenmin-wu closed 3 years ago
Fixed by upgrading the jar version to org.tensorflow:spark-tensorflow-connector_2.11:1.15.0
.
If you follow the example, you may also encounter AttributeError
, this is because the spark uses pickle which cannot pickle the EstimatorWrapper
. To solve this problem, I put the EstimatorWrapper
in a package and upload it to pypi. So we can install the package on both the driver and executors with pip install estimator-wrapper
.
When I follow the document to test the following codes, I came across this Error: Codes:
Error:
spark version: 2.4.3
Conf: