aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 09:44:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix params in decoder (#162)

Browse Source
This commit is contained in:
zion
2022-06-22 16:45:01 -05:00
committed by GitHub
parent c8422ffd5d
commit fddf66e91e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_decoder.py
View File

@@ -258,8 +258,8 @@ def train(
is_master = accelerator.process_index == 0 is_master = accelerator.process_index == 0
trainer = DecoderTrainer( trainer = DecoderTrainer(
accelerator, decoder=decoder,
decoder, accelerator=accelerator,
**kwargs **kwargs
) )