【第6章】Pytorch Lightningの関数定義について(p.82〜83) about bert-book HOT 2 CLOSED

pytorch_lightningがdataloaderからbatchを取り出す時に、batch_idxを自動的に生成し、それをtraining_stepなどの関数に渡していると思われます。実際に、training_stepの中に、print(batch_idx)のコードを入れて、trainer.fit()を実行すると、batch_idxが適切に入力されていることがわかります。

from bert-book.

早速のご回答ありがとうございます。
やはりtrainer.fit()実行時のどこかでbatch_indexは自動的に作られるんですね。
納得しました。

from bert-book.