test

2026-04-09 06:45:09 +00:00 · 2024-01-26 20:02:48 +08:00
parent 85bc371ebc
commit 50f631c768
1 changed files with 2 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -774,7 +774,8 @@ class NetworkTrainer:
                        else:
                            raise NotImplementedError("multipliers for each sample is not supported yet")
                        # print(f"set multiplier: {multipliers}")
-                        print(network)
+                        print(type(network))
+
                        network.set_multiplier(multipliers)

                    with torch.set_grad_enabled(train_text_encoder), accelerator.autocast():