Update notebook

torch2_benchmark.ipynb

@@ -89,200 +89,13 @@
     "mpl.rcParams.update({'font.size': 15})"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 128,
-   "metadata": {
-    "scrolled": true
-   },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['magma',\n",
-       " 'inferno',\n",
-       " 'plasma',\n",
-       " 'viridis',\n",
-       " 'cividis',\n",
-       " 'twilight',\n",
-       " 'twilight_shifted',\n",
-       " 'turbo',\n",
-       " 'Blues',\n",
-       " 'BrBG',\n",
-       " 'BuGn',\n",
-       " 'BuPu',\n",
-       " 'CMRmap',\n",
-       " 'GnBu',\n",
-       " 'Greens',\n",
-       " 'Greys',\n",
-       " 'OrRd',\n",
-       " 'Oranges',\n",
-       " 'PRGn',\n",
-       " 'PiYG',\n",
-       " 'PuBu',\n",
-       " 'PuBuGn',\n",
-       " 'PuOr',\n",
-       " 'PuRd',\n",
-       " 'Purples',\n",
-       " 'RdBu',\n",
-       " 'RdGy',\n",
-       " 'RdPu',\n",
-       " 'RdYlBu',\n",
-       " 'RdYlGn',\n",
-       " 'Reds',\n",
-       " 'Spectral',\n",
-       " 'Wistia',\n",
-       " 'YlGn',\n",
-       " 'YlGnBu',\n",
-       " 'YlOrBr',\n",
-       " 'YlOrRd',\n",
-       " 'afmhot',\n",
-       " 'autumn',\n",
-       " 'binary',\n",
-       " 'bone',\n",
-       " 'brg',\n",
-       " 'bwr',\n",
-       " 'cool',\n",
-       " 'coolwarm',\n",
-       " 'copper',\n",
-       " 'cubehelix',\n",
-       " 'flag',\n",
-       " 'gist_earth',\n",
-       " 'gist_gray',\n",
-       " 'gist_heat',\n",
-       " 'gist_ncar',\n",
-       " 'gist_rainbow',\n",
-       " 'gist_stern',\n",
-       " 'gist_yarg',\n",
-       " 'gnuplot',\n",
-       " 'gnuplot2',\n",
-       " 'gray',\n",
-       " 'hot',\n",
-       " 'hsv',\n",
-       " 'jet',\n",
-       " 'nipy_spectral',\n",
-       " 'ocean',\n",
-       " 'pink',\n",
-       " 'prism',\n",
-       " 'rainbow',\n",
-       " 'seismic',\n",
-       " 'spring',\n",
-       " 'summer',\n",
-       " 'terrain',\n",
-       " 'winter',\n",
-       " 'Accent',\n",
-       " 'Dark2',\n",
-       " 'Paired',\n",
-       " 'Pastel1',\n",
-       " 'Pastel2',\n",
-       " 'Set1',\n",
-       " 'Set2',\n",
-       " 'Set3',\n",
-       " 'tab10',\n",
-       " 'tab20',\n",
-       " 'tab20b',\n",
-       " 'tab20c',\n",
-       " 'magma_r',\n",
-       " 'inferno_r',\n",
-       " 'plasma_r',\n",
-       " 'viridis_r',\n",
-       " 'cividis_r',\n",
-       " 'twilight_r',\n",
-       " 'twilight_shifted_r',\n",
-       " 'turbo_r',\n",
-       " 'Blues_r',\n",
-       " 'BrBG_r',\n",
-       " 'BuGn_r',\n",
-       " 'BuPu_r',\n",
-       " 'CMRmap_r',\n",
-       " 'GnBu_r',\n",
-       " 'Greens_r',\n",
-       " 'Greys_r',\n",
-       " 'OrRd_r',\n",
-       " 'Oranges_r',\n",
-       " 'PRGn_r',\n",
-       " 'PiYG_r',\n",
-       " 'PuBu_r',\n",
-       " 'PuBuGn_r',\n",
-       " 'PuOr_r',\n",
-       " 'PuRd_r',\n",
-       " 'Purples_r',\n",
-       " 'RdBu_r',\n",
-       " 'RdGy_r',\n",
-       " 'RdPu_r',\n",
-       " 'RdYlBu_r',\n",
-       " 'RdYlGn_r',\n",
-       " 'Reds_r',\n",
-       " 'Spectral_r',\n",
-       " 'Wistia_r',\n",
-       " 'YlGn_r',\n",
-       " 'YlGnBu_r',\n",
-       " 'YlOrBr_r',\n",
-       " 'YlOrRd_r',\n",
-       " 'afmhot_r',\n",
-       " 'autumn_r',\n",
-       " 'binary_r',\n",
-       " 'bone_r',\n",
-       " 'brg_r',\n",
-       " 'bwr_r',\n",
-       " 'cool_r',\n",
-       " 'coolwarm_r',\n",
-       " 'copper_r',\n",
-       " 'cubehelix_r',\n",
-       " 'flag_r',\n",
-       " 'gist_earth_r',\n",
-       " 'gist_gray_r',\n",
-       " 'gist_heat_r',\n",
-       " 'gist_ncar_r',\n",
-       " 'gist_rainbow_r',\n",
-       " 'gist_stern_r',\n",
-       " 'gist_yarg_r',\n",
-       " 'gnuplot_r',\n",
-       " 'gnuplot2_r',\n",
-       " 'gray_r',\n",
-       " 'hot_r',\n",
-       " 'hsv_r',\n",
-       " 'jet_r',\n",
-       " 'nipy_spectral_r',\n",
-       " 'ocean_r',\n",
-       " 'pink_r',\n",
-       " 'prism_r',\n",
-       " 'rainbow_r',\n",
-       " 'seismic_r',\n",
-       " 'spring_r',\n",
-       " 'summer_r',\n",
-       " 'terrain_r',\n",
-       " 'winter_r',\n",
-       " 'Accent_r',\n",
-       " 'Dark2_r',\n",
-       " 'Paired_r',\n",
-       " 'Pastel1_r',\n",
-       " 'Pastel2_r',\n",
-       " 'Set1_r',\n",
-       " 'Set2_r',\n",
-       " 'Set3_r',\n",
-       " 'tab10_r',\n",
-       " 'tab20_r',\n",
-       " 'tab20b_r',\n",
-       " 'tab20c_r']"
-      ]
-     },
-     "execution_count": 128,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "list(mpl.colormaps.keys())"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 186,
    "metadata": {},
    "outputs": [],
    "source": [
-    "my_cmap = mpl.colormaps['viridis']   # TODO: use materials, maybe."
+    "# my_cmap = mpl.colormaps['viridis']   # TODO: use materials, maybe."
    ]
   },
   {