fix nonetype reference when ctrl+c

bghira · Sep 14, 2024 · a997ec4 · a997ec4
1 parent 4a633a3
commit a997ec4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/train.py b/train.py
@@ -8,8 +8,6 @@
 logger.setLevel(environ.get("SIMPLETUNER_LOG_LEVEL", "INFO"))
 
 if __name__ == "__main__":
-    global bf
-    bf = None
     trainer = None
     try:
         import multiprocessing
@@ -64,4 +62,4 @@
         print(e)
         print(traceback.format_exc())
     if trainer is not None and trainer.bf is not None:
-        bf.stop_fetching()
+        trainer.bf.stop_fetching()