open graph_kernel flag in mobilenet v3

2021-05-24 10:04:55 +08:00 · 2021-05-24 10:04:55 +08:00 · b5d08986cc
parent 6bbbfe32a5
commit b5d08986cc
1 changed files with 1 additions and 0 deletions
--- a/model_zoo/official/cv/mobilenetv3/train.py
+++ b/model_zoo/official/cv/mobilenetv3/train.py
@ -160,6 +160,7 @@ if __name__ == '__main__':
    config_ = None
    if args_opt.device_target == "GPU":
        config_ = config_gpu
+        context.set_context(enable_graph_kernel=True)
    elif args_opt.device_target == "CPU":
        config_ = config_cpu
    else: