benchmark

killiansheriff · Mar 11, 2024 · c23f861 · c23f861
1 parent c677240
commit c23f861
Showing 1 changed file with 52 additions and 2 deletions.
diff --git a/examples/benchmark.ipynb b/examples/benchmark.ipynb
@@ -10,7 +10,9 @@
     "\n",
     "Note that the time to identify motifs in atomistic data that is spent during dataset creation time could be further reduced if needed. Also note that this dataset creation time is minimal when applied to the synthetic dataset as we only need to permute the atomic type of a reference graph.\n",
     "\n",
-    "Additonally the framework currently relies on pandas dataframe, we may convert it to use polars dataframe instead for speed improvements. \n"
+    "Additonally the framework currently relies on pandas dataframe, we may convert it to use polars dataframe instead for speed improvements. \n",
+    "\n",
+    "We used here an M1 macbook pro, without metal acceleration. \n"
    ]
   },
   {
@@ -109,7 +111,55 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Inference:   0%|          | 0/2 [00:00<?, ?it/s]"
+      "Inference: 100%|██████████| 2/2 [00:19<00:00,  9.77s/it]\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "It took 19.84200633317232s. \n",
+      " \n",
+      "Timing a system with 4000 atoms...\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Inference: 100%|██████████| 16/16 [00:26<00:00,  1.63s/it]\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "It took 26.965233207680285s. \n",
+      " \n",
+      "Timing a system with 13500 atoms...\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Inference: 100%|██████████| 53/53 [00:44<00:00,  1.19it/s]\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "It took 47.50935541698709s. \n",
+      " \n",
+      "Timing a system with 32000 atoms...\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Inference:   0%|          | 0/125 [00:00<?, ?it/s]"
      ]
     }
    ],