fix bug dscnn gpu train failed

2021-09-13 10:02:17 +08:00 · 2021-09-13 10:02:17 +08:00 · da251d5032
parent 901124c9bd
commit da251d5032
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/research/nlp/dscnn/train.py
+++ b/model_zoo/research/nlp/dscnn/train.py
@ -101,7 +101,7 @@ def train():

    # init distributed
    if config.is_distributed:
-        if get_device_id().isdigit():
+        if get_device_id():
            context.set_context(device_id=get_device_id())
        init()
        rank = get_rank_id()