Hi,
I am working on a multi-task model with uneven dataset size and have a custom sampler and using the sampler in dataloader (below)
sampler = BalancedBatchSchedulerSampler(dataset, batch_size)
dataloader = DataLoader(
                dataset,
                sampler=sampler,
                batch_size=batch_size,
                collate_fn=collate_fn,
                num_workers=num_workers,
            )

BalancedBatchSchedulerSampler is the custom sampler. Also I have set replace_sampler_ddp to False. With this custom sampler I don’t see the checkpoint folder is getting created for the model. When I don’t pass the sampler argument and use the default RandomSampler the checkpoint is getting created without any other change in the code.
Is it possible that the sampler is affecting the model checkpoint somehow?
Thank you!