(examples)
| 188 | ) |
| 189 | |
| 190 | def collate_fn_ours(examples): |
| 191 | pixel_values = torch.stack([example["image"] for example in examples]) |
| 192 | pixel_values = pixel_values.to(memory_format=torch.contiguous_format).float() |
| 193 | batch = {"pixel_values": pixel_values} |
| 194 | return batch |
| 195 | |
| 196 | datasets_st = AnyWordLmdbDataset( |
| 197 | lmdb_path=cfgs.lmdb_path, |