Code

PyTorch model extractor#298

Code

Open

mastoffel wants to merge 6 commits intomainalan-turing-institute/autoemulate:mainfrom pytorchalan-turing-institute/autoemulate:pytorchCopy head branch name to clipboard

+171

Commits

Commits on Feb 7, 2025

• add is_fitted attribute to cnp and attn_cnp

  mastoffel

  committed
  507155a

  Copy full SHA for 507155a

• add pytorch extractor+tests

  mastoffel

  committed
  c0c85aa

  Copy full SHA for c0c85aa

• fix pytorch extractor tests

  mastoffel

  committed
  4ea7446

  Copy full SHA for 4ea7446

Commits on Feb 11, 2025

• add warning that scalers are removed

  mastoffel

  committed
  c456c38

  Copy full SHA for c456c38

Commits on Feb 12, 2025

• refactor extract pytorch model

  mastoffel

  committed
  ce18abe

  Copy full SHA for ce18abe

• return pytorch model in eval mode

  mastoffel

  committed
  40dbed2

  Copy full SHA for 40dbed2