edit dev notebook

notebooks/dev.ipynb

@@ -45,7 +45,7 @@
     "\n",
     "actions = build_actions_tensor(size)\n",
     "\n",
-    "cube = Cube.create([\"U\", \"L\", \"C\", \"R\", \"B\", \"D\"], size=size)\n",
+    "cube = Cube([\"U\", \"L\", \"C\", \"R\", \"B\", \"D\"], size=size)\n",
     "print(cube)"
    ]
   },
@@ -56,10 +56,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "cubis = copy.deepcopy(cube)\n",
-    "cubis.shuffle(5000, seed=0)\n",
-    "print(cubis)\n",
-    "print(cubis.history)"
+    "cubis_cpu = copy.deepcopy(cube)\n",
+    "cubis_cpu.shuffle(2000, seed=0)\n",
+    "print(cubis_cpu)\n",
+    "print(cubis_cpu.history)"
    ]
   },
   {
@@ -82,7 +82,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "(actions[0, 2, 0].type(torch.FloatTensor) @ actions[0, 1, 1].type(torch.FloatTensor)).type(torch.int8)"
+    "(actions[0, 2, 0].type(torch.float32) @ actions[0, 1, 1].type(torch.float32)).type(torch.int8)"
    ]
   }
  ],