Commit 88b3c88a authored by lucas_miranda's avatar lucas_miranda
Browse files

Added a MirroredStrategy to train models on multiple GPUs if they are available

parent 23ff6d44
Pipeline #100287 passed with stages
in 19 minutes and 12 seconds
......@@ -23,7 +23,7 @@ setuptools.setup(
License="MIT",
include_package_data=True,
install_requires=[
"tensorflow",
"tensorflow>=2.4",
"numpy",
"pandas",
"joblib",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment