Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Custom torch.utils.data.DataLoader.collate_fn #1328

Open
fabriceyhc opened this issue Jul 17, 2021 · 1 comment
Open

Custom torch.utils.data.DataLoader.collate_fn #1328

fabriceyhc opened this issue Jul 17, 2021 · 1 comment

Comments

@fabriceyhc
Copy link

I'm working on a series of data augmentations for text and have implemented their use via the torch.utils.data.DataLoader.collate_fn but jiant appears to be using this function already. Is there a recommended workaround or roadmap item to accommodate custom a collate_fn alongside your own? If not, I'll just work on my own changes and possibly submit as a pull request.
@zphang
Copy link
Collaborator

zphang commented Jul 26, 2021

Hi,

Not as far as I know. Feel free to make a PR!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@zphang @fabriceyhc