Code

Added support for ensemble models#308

Code

Merged

stefdoerr merged 12 commits intomaintorchmd/torchmd-net:mainfrom ensemble_model_supporttorchmd/torchmd-net:ensemble_model_supportCopy head branch name to clipboardMar 19, 2024

+175-62

Commits

Commits on Mar 18, 2024

• Added support for ensemble models

  stefdoerr

  committed
  1b67010

  Copy full SHA for 1b67010

• remove debug print

  stefdoerr

  committed
  a5c0a3a

  Copy full SHA for a5c0a3a

• Update load_model docstring

  RaulPPelaez

  committed
  2b99e77

  Copy full SHA for 2b99e77

• Update Ensemble docstring

  RaulPPelaez

  committed
  d071cd3

  Copy full SHA for d071cd3

• Add test for ensemble

  RaulPPelaez

  committed
  94f94d9

  Copy full SHA for 94f94d9

Commits on Mar 19, 2024

• returning the standard deviation is now optional

  stefdoerr

  committed
  5ba3db9

  Copy full SHA for 5ba3db9

• fix test

  stefdoerr

  committed
  417955d

  Copy full SHA for 417955d

• Merge branch 'main' into ensemble_model_support

  RaulPPelaez

  authored
  9b0a362

  Copy full SHA for 9b0a362

• Make Ensemble variadic

  RaulPPelaez

  committed
  5a60b40

  Copy full SHA for 5a60b40

• Update documentation

  RaulPPelaez

  committed
  013a691

  Copy full SHA for 013a691

• Update docstrings

  RaulPPelaez

  committed
  365edea

  Copy full SHA for 365edea

• Update docstring

  RaulPPelaez

  committed
  f21f3de

  Copy full SHA for f21f3de