Skip to content

Commit

Permalink
Updating Tensorflow version to 2.17, reducing training rounds to 3
Browse files Browse the repository at this point in the history
Signed-off-by: Das, Snehal <[email protected]>
  • Loading branch information
snehal-das committed Nov 22, 2024
1 parent 63ffa8b commit 697408b
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions openfl-tutorials/experimental/104_Keras_MNIST_with_CPU.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -38,15 +38,15 @@
"source": [
"#Install openfl and required packages for the workflow APIs to function\n",
"%pip install git+https://github.com/securefederatedai/openfl.git\n",
"%pip install -r workflow_interface_requirements.txt\n",
"\n",
"# Uncomment this if running in Google Colab and set USERNAME if running in docker container.\n",
"#%pip install -r https://raw.githubusercontent.com/intel/openfl/develop/openfl-tutorials/experimental/workflow_interface_requirements.txt\n",
"#import os\n",
"#os.environ[\"USERNAME\"] = \"colab\"\n",
"#%pip install -r workflow_interface_requirements.txt\n",
"\n",
"#Install Tensorflow to access Keras\n",
"%pip install tensorflow==2.13"
"%pip install tensorflow==2.17\n",
"\n",
"# Uncomment this if running in Google Colab and set USERNAME if running in docker container.\n",
"%pip install -r https://raw.githubusercontent.com/intel/openfl/develop/openfl-tutorials/experimental/workflow_interface_requirements.txt\n",
"import os\n",
"os.environ[\"USERNAME\"] = \"colab\""
]
},
{
Expand Down Expand Up @@ -346,7 +346,7 @@
"metadata": {},
"outputs": [],
"source": [
"flflow = KerasMNISTWorkflow(model, rounds=10, checkpoint=True)\n",
"flflow = KerasMNISTWorkflow(model, rounds=3, checkpoint=True)\n",
"flflow.runtime = local_runtime\n",
"flflow.run()"
]
Expand Down

0 comments on commit 697408b

Please sign in to comment.