mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-09 06:45:09 +00:00
add debug log
This commit is contained in:
@@ -354,6 +354,16 @@ class NetworkTrainer:
|
||||
trainable_params = network.prepare_optimizer_params(args.text_encoder_lr, args.unet_lr)
|
||||
lr_descriptions = None
|
||||
|
||||
# if len(trainable_params) == 0:
|
||||
# accelerator.print("no trainable parameters found / 学習可能なパラメータが見つかりませんでした")
|
||||
# for params in trainable_params:
|
||||
# for k, v in params.items():
|
||||
# if type(v) == float:
|
||||
# pass
|
||||
# else:
|
||||
# v = len(v)
|
||||
# accelerator.print(f"trainable_params: {k} = {v}")
|
||||
|
||||
optimizer_name, optimizer_args, optimizer = train_util.get_optimizer(args, trainable_params)
|
||||
|
||||
# dataloaderを準備する
|
||||
|
||||
Reference in New Issue
Block a user