Final pass removing dataset dependence from disentangling notebook.

PiperOrigin-RevId: 321177642
google · Jul 14, 2020 · a2ad408 · a2ad408
1 parent 8a994ef
commit a2ad408
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/Disentangling_Trainability_and_Generalization.ipynb b/notebooks/Disentangling_Trainability_and_Generalization.ipynb
@@ -908,7 +908,7 @@
       },
       "outputs": [],
       "source": [
-        "x_train, y_train, x_test, y_test = normalize(*datasets.get_dataset('cifar10', 64, 32))"
+        "x_train, y_train, x_test, y_test = normalize(*get_dataset('cifar10', 64, 32))"
       ]
     },
     {
@@ -1417,7 +1417,7 @@
       },
       "outputs": [],
       "source": [
-        "x_train, y_train, x_test, y_test = tuple(np.array(x) for x in datasets.get_dataset('cifar10', 256, 10000))"
+        "x_train, y_train, x_test, y_test = tuple(np.array(x) for x in get_dataset('cifar10', 256, 10000))"
       ]
     },
     {