metadata |= to metadata.update

This commit is contained in:
camenduru
2023-03-06 00:31:28 +03:00
committed by GitHub
parent 46aee85d2a
commit 772ee52ef2

View File

@@ -408,7 +408,7 @@ def train(args):
"img_count": subset.img_count
}
metadata |= {
metadata.update({
"ss_batch_size_per_device": args.train_batch_size,
"ss_total_batch_size": total_batch_size,
"ss_resolution": args.resolution,
@@ -425,7 +425,7 @@ def train(args):
"ss_reg_dataset_dirs": json.dumps(reg_dataset_dirs_info),
"ss_tag_frequency": json.dumps(dataset.tag_frequency),
"ss_bucket_info": json.dumps(dataset.bucket_info),
}
})
# uncomment if another network is added
# for key, value in net_kwargs.items():