|
@@ -285,19 +285,22 @@
|
|
|
},
|
|
|
{
|
|
|
"cell_type": "code",
|
|
|
- "execution_count": 20,
|
|
|
+ "execution_count": null,
|
|
|
"metadata": {},
|
|
|
"outputs": [],
|
|
|
"source": [
|
|
|
"train_images = processed_train_data.shape[0]\n",
|
|
|
+ "print(train_images)\n",
|
|
|
"train_indexes = list(range(train_images))\n",
|
|
|
"random.shuffle(train_indexes)\n",
|
|
|
"\n",
|
|
|
"test_images = processed_test_data.shape[0]\n",
|
|
|
+ "print(test_images)\n",
|
|
|
"test_indexes = list(range(test_images))\n",
|
|
|
"random.shuffle(test_indexes)\n",
|
|
|
"\n",
|
|
|
"val_images = processed_val_data.shape[0]\n",
|
|
|
+ "print(val_images)\n",
|
|
|
"val_indexes = list(range(val_images))\n",
|
|
|
"random.shuffle(val_indexes)\n",
|
|
|
"\n",
|
|
@@ -383,7 +386,7 @@
|
|
|
],
|
|
|
"metadata": {
|
|
|
"kernelspec": {
|
|
|
- "display_name": "Python 3",
|
|
|
+ "display_name": "groupvit",
|
|
|
"language": "python",
|
|
|
"name": "python3"
|
|
|
},
|