VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix wrong typing

Browse Source
This commit is contained in:
Isotr0py
2023-03-09 19:35:06 +08:00
parent eb68892ab1
commit ab05be11d2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/train_util.py
View File

@@ -1877,7 +1877,7 @@ def get_scheduler_fix(args,optimizer: Optimizer):
power = args.lr_scheduler_power power = args.lr_scheduler_power
lr_scheduler_kwargs = {} # get custom lr_scheduler kwargs lr_scheduler_kwargs = {} # get custom lr_scheduler kwargs
if args.lr_scheduler_args is not None and len(args.optimizer_args) > 0: if args.lr_scheduler_args is not None and len(args.lr_scheduler_args) > 0:
for arg in args.lr_scheduler_args: for arg in args.lr_scheduler_args:
key, value = arg.split('=') key, value = arg.split('=')
@@ -1886,11 +1886,11 @@ def get_scheduler_fix(args,optimizer: Optimizer):
if value[i].lower() == "true" or value[i].lower() == "false": if value[i].lower() == "true" or value[i].lower() == "false":
value[i] = (value[i].lower() == "true") value[i] = (value[i].lower() == "true")
else: else:
value[i] = float(value[i]) value[i] = eval(value[i]) # warning: not safe!!!
if len(value) == 1: if len(value) == 1:
value = value[0] value = value[0]
else: else:
value = tuple(value) value = tuple(value) # some may use list?
lr_scheduler_kwargs[key] = value lr_scheduler_kwargs[key] = value