iicd/ddpo-pytorch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Device specific seed

Browse Source
This commit is contained in:
Kevin Black
2023-06-26 22:35:24 -07:00
parent 1ce0994c8a
commit 4c5322ca85
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/train.py
View File

@@ -45,7 +45,7 @@ def main(_):
logger.info(config) logger.info(config)
# set seed # set seed
set_seed(config.seed) set_seed(config.seed, device_specific=True)
# load scheduler, tokenizer and models. # load scheduler, tokenizer and models.
pipeline = StableDiffusionPipeline.from_pretrained(config.pretrained.model, revision=config.pretrained.revision) pipeline = StableDiffusionPipeline.from_pretrained(config.pretrained.model, revision=config.pretrained.revision)