Update rwkv_encoder.py

aky15

2024-01-11 c7c97cb0fe91b06dcbc391b5a7dee92d496dc894

 funasr/datasets/dataloader_fn.py

@@ -38,13 +38,16 @@
batch_sampler = BatchSampler(dataset)


def collator(samples: list = None):
    return samples

if __name__ == "__main__":
    
    dataloader_tr = torch.utils.data.DataLoader(dataset,
                                                collate_fn=dataset.collator,
                                                batch_sampler=batch_sampler,
                                                shuffle=False,
                                                num_workers=0,
                                                num_workers=8,
                                                pin_memory=True)
    
    print(len(dataset))