Add MEGNet formation energy model example.

materialsvirtuallab · Jun 20, 2023 · f92699c · f92699c
1 parent e4c8b0d
commit f92699c
Show file tree

Hide file tree

Showing 2 changed files with 361 additions and 1 deletion.
diff --git a/examples/Training a M3GNet Potential with PyTorch Lightning.ipynb b/examples/Training a M3GNet Potential with PyTorch Lightning.ipynb
@@ -144,7 +144,7 @@
    "id": "01be4689",
    "metadata": {},
    "source": [
-    "Finally, we will initialize the Pytorch Lightning trainer and run the fitting. Here, the max_epochs is set to 2 just for demonstration purposes. In a real fitting, this would be a much larger number. Also, the `accelerator` "
+    "Finally, we will initialize the Pytorch Lightning trainer and run the fitting. Here, the max_epochs is set to 2 just for demonstration purposes. In a real fitting, this would be a much larger number. Also, the `accelerator=\"cpu\"` was set just to ensure compatibility with M1 Macs. In a real world use case, please remove the kwarg or set it to cuda for GPU based training. "
    ]
   },
   {