Skip to content

GitLab

Unverified Commit cfd1d5ee authored by Haofan Wang's avatar Haofan Wang Committed by GitHub
Browse files

[example] fixed seed error in train_dreambooth_colossalai.py (#2445)

parent ac18a445
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
examples/images/dreambooth/train_dreambooth_colossalai.py
View file @ cfd1d5ee
...@@ -355,10 +355,11 @@ def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "auto"): ...@@ -355,10 +355,11 @@ def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "auto"):
def main(args): def main(args):
colossalai.launch_from_torch(config={})
if args.seed is not None: if args.seed is None:
gpc.set_seed(args.seed) colossalai.launch_from_torch(config={})
else:
colossalai.launch_from_torch(config={}, seed=args.seed)
if args.with_prior_preservation: if args.with_prior_preservation:
class_images_dir = Path(args.class_data_dir) class_images_dir = Path(args.class_data_dir)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment