Skip to content

Conversation

dansuissa
Copy link
Collaborator

I fixed the NotImplementedError in the IterableJSONData class by implementing a proper iterator that correctly reads and processes JSON files line by line in distributed training environments. I also solved the total_steps calculation for the OneCycleLR scheduler by adding a method that accurately computes steps based on dataset size, batch size, number of GPUs, and gradient accumulation settings.

@Adibvafa
Copy link
Owner

Adibvafa commented Jul 1, 2025

@dansuissa Can you write and run tests, also add explanation to README

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants