yfeng24816

Finetune Transformers Models with PyTorch Lightning: documentation error?

4

When calculating the total steps, shouldn't we use `number of batches * epoch size` ? In this case, it would be `self.total_steps = (len(train_loader.dataset) // tb_size) * ab_size` instead of...