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

pydantic 2

Browse Source
This commit is contained in:
Phil Wang
2023-07-15 09:32:44 -07:00
parent 00e07b7d61
commit 40843bcc21
4 changed files with 19 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
train_decoder.py
View File

@@ -577,6 +577,7 @@ def initialize_training(config: TrainDecoderConfig, config_path):
shards_per_process = len(all_shards) // world_size
assert shards_per_process > 0, "Not enough shards to split evenly"
my_shards = all_shards[rank * shards_per_process: (rank + 1) * shards_per_process]
dataloaders = create_dataloaders (
available_shards=my_shards,
img_preproc = config.data.img_preproc,