We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a6c5ea7 commit 1645017Copy full SHA for 1645017
1 file changed
train_network.py
@@ -893,6 +893,9 @@ def train(self, args):
893
if args.base_weights is not None and not isinstance(args.base_weights, list):
894
args.base_weights = [args.base_weights]
895
896
+ if args.base_weights_multiplier is not None and not isinstance(args.base_weights_multiplier, list):
897
+ args.base_weights_multiplier = [args.base_weights_multiplier]
898
+
899
if args.base_weights is not None:
900
# base_weights が指定されている場合は、指定された重みを読み込みマージする
901
for i, weight_path in enumerate(args.base_weights):
0 commit comments