Code

add MPS support to base_trainer.py#151

Code

Open

francescomalandrino wants to merge 1 commit intoclementchadebec:mainclementchadebec/benchmark_VAE:mainfrom francescomalandrino:rudimentary-mpsfrancescomalandrino/benchmark_VAE:rudimentary-mpsCopy head branch name to clipboard

+9-7

Commits

Commits on Jul 31, 2024

• Update base_trainer.py

  Show description for c1a72b6
  francescomalandrino

  authored
  c1a72b6

  Copy full SHA for c1a72b6