test

2026-04-08 22:35:09 +00:00 · 2024-01-26 18:58:47 +08:00
parent c576f80639
commit 85bc371ebc
1 changed files with 1 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -774,6 +774,7 @@ class NetworkTrainer:
                        else:
                            raise NotImplementedError("multipliers for each sample is not supported yet")
                        # print(f"set multiplier: {multipliers}")
+                        print(network)
                        network.set_multiplier(multipliers)

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