From f43b2b82fbbb8036436e24a0500156f81e0de6ea Mon Sep 17 00:00:00 2001 From: Dean of NITSC <138109564+nitsc@users.noreply.github.com> Date: Tue, 13 Aug 2024 11:49:10 +0800 Subject: [PATCH] Update run.py --- run.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/run.py b/run.py index 31bc6da..95d679c 100644 --- 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()