Finetune Transformers Models with PyTorch Lightning: documentation error?

[yfeng24816]

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 self.total_steps = (len(train_loader.dataset) // tb_size) // ab_size.

Please fix me if anywhere is wrong.

https://pytorchlightning.github.io/lightning-tutorials/notebooks/lightning_examples/text-transformers.html

cc @borda @rohitgr7

[rohitgr7]

I guess it should be batches * num_epochs but why would it be * ab_size?

[yfeng24816]

Is ab_size something like num_epochs? It becomes self.trainer.max_epochs when accumulate_grad_batches is 1.

[rohitgr7]

okay. yes... I didn't see max_epochs there. should be something like

total = (total / accumulation_factor)*max_epochs.

[yfeng24816]

So do you also think there is an error in the documentation too? I am not sure by my own.