turn use_cuda=False when do testing

61d08328 · Zeyu Chen · 074d166d · 61d08328
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tests/test_export_n_load_module.py tests/test_export_n_load_module.py +1 -1

未找到文件。
--- a/tests/test_export_n_load_module.py
+++ b/tests/test_export_n_load_module.py
@@ -234,7 +234,7 @@ def test_load_module(use_cuda=False):


 if __name__ == "__main__":
-    use_cuda = True
+    use_cuda = False
    print("train...")
    train(use_cuda)
    print("save module...")