Update run.py

2024-08-13 11:49:10 +08:00 · 2024-08-13 11:49:10 +08:00 · f43b2b82fb
parent c6ebcadb0b
commit f43b2b82fb
1 changed files with 3 additions and 0 deletions
--- a/run.py
+++ b/run.py
@ -2,5 +2,8 @@

 from modules import core

+import os
+os.environ['KERAS_BACKEND'] = 'tensorflow'
+
 if __name__ == '__main__':
    core.run()