From 75a5bb17ba1fa39c15bc2c5e36801092d9232d8b Mon Sep 17 00:00:00 2001 From: baocheny Date: Tue, 27 Jun 2023 10:58:32 +0800 Subject: [PATCH] add 2 more custom devices intel_gpu and apple mps --- ppcls/engine/engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index da829282..3a8ebdef 100755 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -101,7 +101,7 @@ class Engine(object): # set device assert self.config["Global"][ - "device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend"] + "device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend", "intel_gpu", "mps"] self.device = paddle.set_device(self.config["Global"]["device"]) logger.info('train with paddle {} and device {}'.format( paddle.__version__, self.device)) -- GitLab