From 0ccab14c5ad12e098622773d37b493d2f9a8c066 Mon Sep 17 00:00:00 2001
From: GhostScreaming <mofengshenjieII@163.com>
Date: Tue, 8 Nov 2022 08:15:59 +0000
Subject: [PATCH] Set flag FLAGS_enable_cublas_tensor_op_math on in default.

---
 tools/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tools/train.py b/tools/train.py
index 9fdf56cad..b28f8c547 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -42,6 +42,7 @@ def set_default_flags(flags):
 
 
 if __name__ == "__main__":
+    set_default_flags({'FLAGS_enable_cublas_tensor_op_math': True, })
     args = config.parse_args()
     cfg = config.get_config(args.config, overrides=args.override, show=False)