Updated Transformer doc notebooks with commit b5e2b183af5e40e33a4dc7659e697d137259d56e \n\nSee: huggingface/transformers@b5e2b18

Author: Hugging Face

transformers_doc/benchmarks.ipynb

@@ -72,7 +72,9 @@
    "source": [
     "from transformers import TensorFlowBenchmark, TensorFlowBenchmarkArguments\n",
     "\n",
-    "args = TensorFlowBenchmarkArguments(models=[\"bert-base-uncased\"], batch_sizes=[8], sequence_lengths=[8, 32, 128, 512])\n",
+    "args = TensorFlowBenchmarkArguments(\n",
+    "    models=[\"bert-base-uncased\"], batch_sizes=[8], sequence_lengths=[8, 32, 128, 512]\n",
+    ")\n",
     "benchmark = TensorFlowBenchmark(args)"
    ]
   },
@@ -322,7 +324,9 @@
    "source": [
     "from transformers import PyTorchBenchmark, PyTorchBenchmarkArguments, BertConfig\n",
     "\n",
-    "args = PyTorchBenchmarkArguments(models=[\"bert-base\", \"bert-384-hid\", \"bert-6-lay\"], batch_sizes=[8], sequence_lengths=[8, 32, 128, 512])\n",
+    "args = PyTorchBenchmarkArguments(\n",
+    "    models=[\"bert-base\", \"bert-384-hid\", \"bert-6-lay\"], batch_sizes=[8], sequence_lengths=[8, 32, 128, 512]\n",
+    ")\n",
     "config_base = BertConfig()\n",
     "config_384_hid = BertConfig(hidden_size=384)\n",
     "config_6_lay = BertConfig(num_hidden_layers=6)\n",
@@ -408,7 +412,9 @@
    "source": [
     "from transformers import TensorFlowBenchmark, TensorFlowBenchmarkArguments, BertConfig\n",
     "\n",
-    "args = TensorFlowBenchmarkArguments(models=[\"bert-base\", \"bert-384-hid\", \"bert-6-lay\"], batch_sizes=[8], sequence_lengths=[8, 32, 128, 512])\n",
+    "args = TensorFlowBenchmarkArguments(\n",
+    "    models=[\"bert-base\", \"bert-384-hid\", \"bert-6-lay\"], batch_sizes=[8], sequence_lengths=[8, 32, 128, 512]\n",
+    ")\n",
     "config_base = BertConfig()\n",
     "config_384_hid = BertConfig(hidden_size=384)\n",
     "config_6_lay = BertConfig(num_hidden_layers=6)\n",