From 50f631c768eb3db578c98224106236cf3abc03f9 Mon Sep 17 00:00:00 2001
From: DukeG <fireicewolf@gmail.com>
Date: Fri, 26 Jan 2024 20:02:48 +0800
Subject: [PATCH] test

---
 train_network.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train_network.py b/train_network.py
index 2e06262f..d64d7a1b 100644
--- 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():