updating notebooks to include the OMP_NUM_THREADS environment variable; DRAMATIC SPEED UP

day_1/0_Using_Pretrained_Embeddings.ipynb

@@ -6,11 +6,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from argparse import Namespace\n",
+    "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
+    "\n",
     "from annoy import AnnoyIndex\n",
     "import numpy as np\n",
     "import torch\n",
-    "from tqdm import tqdm_notebook\n",
-    "from argparse import Namespace"
+    "from tqdm import tqdm_notebook\n"
    ]
   },
   {
@@ -529,7 +532,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.0"
+   "version": "3.6.8"
   }
  },
  "nbformat": 4,

day_1/1_PyTorch_Basics.ipynb

@@ -24,6 +24,9 @@
     }
    ],
    "source": [
+    "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
+    "\n",
     "import torch\n",
     "import numpy as np\n",
     "torch.manual_seed(1234)"
@@ -2758,7 +2761,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.0"
+   "version": "3.6.8"
   }
  },
  "nbformat": 4,

day_1/2_classify_names_with_MLP.ipynb

@@ -14,6 +14,8 @@
    "outputs": [],
    "source": [
     "from argparse import Namespace\n",
+    "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
@@ -1044,15 +1046,6 @@
     "## Evaluate on test set"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 39,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "args.device = \"cuda\""
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 40,

day_1/3_Chinese_document_classification.ipynb

@@ -32,6 +32,7 @@
     "from argparse import Namespace\n",
     "from collections import Counter\n",
     "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/00-Dataset-Loading-And-Vectorization.ipynb

@@ -9,6 +9,8 @@
     "from argparse import Namespace\n",
     "from collections import Counter\n",
     "import json\n",
+    "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import numpy as np\n",
     "import pandas as pd\n",

day_2/01-Char-RNN-Classify-Surnames.ipynb

@@ -49,6 +49,7 @@
     "from collections import Counter\n",
     "import json\n",
     "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/02-Char-RNN-Predict-Surnames.ipynb

@@ -49,6 +49,7 @@
     "from collections import Counter\n",
     "import json\n",
     "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/03-Char-RNN-Conditionally-Predict-Surnames.ipynb

@@ -49,6 +49,7 @@
     "from collections import Counter\n",
     "import json\n",
     "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/CYOA-Amazon-Reviews.ipynb

@@ -26,6 +26,7 @@
     "import json\n",
     "import os\n",
     "import re\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/CYOA-CFPB-Classification.ipynb

@@ -26,6 +26,7 @@
     "import json\n",
     "import os\n",
     "import re\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/CYOA-Surname-Classification.ipynb

@@ -49,6 +49,7 @@
     "from collections import Counter\n",
     "import json\n",
     "import os\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",

day_2/CYOA-Twitter-Language-ID.ipynb

@@ -95,6 +95,7 @@
     "import json\n",
     "import os\n",
     "import re\n",
+    "os.environ['OMP_NUM_THREADS'] = '4' \n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",