Log accelerator device

kohya-ss · Jan 18, 2024 · 34d5260 · 34d5260
1 parent 9fbff99
commit 34d5260
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -3906,6 +3906,7 @@ def prepare_accelerator(args: argparse.Namespace):
         kwargs_handlers=kwargs_handlers,
         dynamo_backend=dynamo_backend,
     )
+    print("accelerator device:", accelerator.device)
     return accelerator