add 2 more custom devices intel_gpu and apple mps

3d0c0eb5 · baocheny · cuicheng01 · 62c56ce2 · 3d0c0eb5
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

ppcls/static/train.py ppcls/static/train.py +3 -2

未找到文件。
--- a/ppcls/static/train.py
+++ b/ppcls/static/train.py
@@ -90,8 +90,9 @@ def main(args):
        fleet.init(is_collective=True)

    # assign the device
-    assert global_config[
-        "device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend"]
+    assert global_config["device"] in [
+        "cpu", "gpu", "xpu", "npu", "mlu", "ascend", "intel_gpu", "mps"
+    ]
    device = paddle.set_device(global_config["device"])

    # amp related config