Use MPS backend if available and use_device=None
, add CHGNET_DEVICE
env var
#302
Job | Run time |
---|---|
2m 40s | |
8s | |
2m 48s |
use_device=None
, add CHGNET_DEVICE
env var
#302
Job | Run time |
---|---|
2m 40s | |
8s | |
2m 48s |