github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-25 15:01:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Haofan Wang 2023-01-11 16:56:15 +08:00 committed by GitHub
parent ac18a445fa
commit cfd1d5ee49
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/images/dreambooth/train_dreambooth_colossalai.py
View File

@ -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)