Commit 5ba66888 authored by Cagri Eryilmaz's avatar Cagri Eryilmaz
Browse files

missing zero in nfnet notebook

parent 886d0b3d
...@@ -213,7 +213,7 @@ ...@@ -213,7 +213,7 @@
"# Run the model\n", "# Run the model\n",
"start = time.time()\n", "start = time.time()\n",
"results = model.run({'inputs':data_input}) # Your first inference would take longer than the following ones.\n", "results = model.run({'inputs':data_input}) # Your first inference would take longer than the following ones.\n",
"print(f\"Time inference took: {100*(time.time() - start):.2f}ms\")\n", "print(f\"Time inference took: {1000*(time.time() - start):.2f}ms\")\n",
"# Extract the index of the top prediction\n", "# Extract the index of the top prediction\n",
"res_npa = np.array(results[0])\n", "res_npa = np.array(results[0])\n",
"print(f\"\\nResult: {labels[np.argmax(res_npa)]}\")" "print(f\"\\nResult: {labels[np.argmax(res_npa)]}\")"
...@@ -228,7 +228,7 @@ ...@@ -228,7 +228,7 @@
"# Run the model again, first one would take long\n", "# Run the model again, first one would take long\n",
"start = time.time()\n", "start = time.time()\n",
"results = model.run({'inputs':data_input}) # Your first inference would take longer than the following ones.\n", "results = model.run({'inputs':data_input}) # Your first inference would take longer than the following ones.\n",
"print(f\"Time inference took: {100*(time.time() - start):.2f}ms\")\n", "print(f\"Time inference took: {1000*(time.time() - start):.2f}ms\")\n",
"# Extract the index of the top prediction\n", "# Extract the index of the top prediction\n",
"res_npa = np.array(results[0])\n", "res_npa = np.array(results[0])\n",
"print(f\"\\nResult: {labels[np.argmax(res_npa)]}\")" "print(f\"\\nResult: {labels[np.argmax(res_npa)]}\")"
......
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