Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

incompatibility with TF2.3 keras optimizers #39

Open
ra312 opened this issue Aug 25, 2020 · 0 comments
Open

incompatibility with TF2.3 keras optimizers #39

ra312 opened this issue Aug 25, 2020 · 0 comments

Comments

@ra312
Copy link

ra312 commented Aug 25, 2020

When initializing SparkAsyncDL with optimizer='adam', we get the following exception
Traceback (most recent call last):
File "spark-train.py", line 170, in
p = Pipeline(stages=[va, spark_model]).fit(train)
File "/usr/local/spark/python/pyspark/ml/base.py", line 132, in fit
return self._fit(dataset)
File "/usr/local/spark/python/pyspark/ml/pipeline.py", line 109, in _fit
model = stage.fit(dataset)
File "/usr/local/spark/python/pyspark/ml/base.py", line 132, in fit
return self._fit(dataset)
File "/data/#user_folder#/sparkflow/sparkflow/tensorflow_async.py", line 278, in _fit
tf_optimizer = build_optimizer(self.getTfOptimizer(), self.getTfLearningRate(), optimizer_options)

tensorflow.version = 2.3.0
sparkflow = 0.7.0
File "/data/#user_folder#/sparkflow/sparkflow/tensorflow_async.py", line 20, in build_optimizer
'adam': tf.train.AdamOptimizer,
AttributeError: module 'tensorflow._api.v2.train' has no attribute 'AdamOptimizer'

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant