fix typo

d70668cb · kuizhiqing · Wei Shengyu · a37cee70 · d70668cb
显示空白变更内容
内联并排

Showing with 1 addition and 2 deletion

ppcls/arch/backbone/legendary_models/swin_transformer.py ppcls/arch/backbone/legendary_models/swin_transformer.py +1 -2

未找到文件。
--- a/ppcls/arch/backbone/legendary_models/swin_transformer.py
+++ b/ppcls/arch/backbone/legendary_models/swin_transformer.py
@@ -86,8 +86,7 @@ class RollWrapperSingleton():
    def __init__(self):
        self.index_dict = {}
        self.roll_with_index_select = RollWithIndexSelect.apply
-        if 'npu' in paddle.device.get_all_custom_device_type():
-            self.enable = True
+        self.enable = True if 'npu' in paddle.device.get_all_custom_device_type() else False

    def __call__(self, x, shifts, axis):
        if not self.enable: