Skip to content
New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

How can we access the input_ids/attention mask in each train batch loop? #42

Open
tkmaker opened this issue Jul 2, 2022 · 0 comments
Open

How can we access the input_ids/attention mask in each train batch loop? #42

tkmaker opened this issue Jul 2, 2022 · 0 comments

Comments

@tkmaker
Copy link

tkmaker commented Jul 2, 2022 •
edited
Loading

I tried using a train step callback but I am not sure how to get access to the dataloader input_ids and attention mask during each train step. Is this possible?

BTW Thanks for the library!
# for free to join this conversation on GitHub. Already have an account? # to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@tkmaker