Use MPS backend if available and use_device=None
, add CHGNET_DEVICE
env var
#301
Job | Run time |
---|---|
3m 30s | |
11s | |
3m 41s |
use_device=None
, add CHGNET_DEVICE
env var
#301
Job | Run time |
---|---|
3m 30s | |
11s | |
3m 41s |