Code

pt: explicitly set device#3307

Code

Merged

njzjz merged 8 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom njzjz:pt-set-devicenjzjz/deepmd-kit:pt-set-deviceCopy head branch name to clipboardFeb 21, 2024

+284-204

Commits

Commits on Feb 20, 2024

• pt: explicitly set device

  Show description for dc8d06e
  njzjz

  committed
  dc8d06e

  Copy full SHA for dc8d06e

• fix trainer.get_data

  Show description for 51ea91e
  njzjz

  committed
  51ea91e

  Copy full SHA for 51ea91e

• set cpu

  Show description for dbd567a
  njzjz

  committed
  dbd567a

  Copy full SHA for dbd567a

• Merge branch 'devel' into pt-set-device

  njzjz

  committed
  3400329

  Copy full SHA for 3400329

• fix test_polar

  Show description for 884b566
  njzjz

  committed
  884b566

  Copy full SHA for 884b566

• fix tests

  Show description for 77231db
  njzjz

  committed
  77231db

  Copy full SHA for 77231db

Commits on Feb 21, 2024

• Merge branch 'devel' into pt-set-device

  Show description for 2958b73
  njzjz

  authored
  2958b73

  Copy full SHA for 2958b73

• fix new as_tensor

  Show description for 6ba2914
  njzjz

  committed
  6ba2914

  Copy full SHA for 6ba2914