Does FSDP support uneven inputs like DDP using Join Hook

Does FSDP support uneven inputs like DDP using Join Hook

It does not support the join hook.

We have found that most people solve the uneven-batch-size-per-rank issue at the dataloader level now.

2 Likes

What’s the most common method of resolving it at the dataloader level?

The dataloader ensures every rank gets the same local batch size.

2 Likes

Thanks for your reply.