training is so slow

[tengshaofeng]

Hi, You really did a great job. thanks so much. When I train it myself. It is so slow, I do not think it is the right speed.
Image gives the trainning process. How do you think that?

Because I use the pytorch-lightning==2.2.3, I can run only when I modified from "pl.Trainer(gpus=1, precision=32, callbacks=[logger])" to "pl.Trainer(precision=32, callbacks=[logger])". Dose it matter?

[tengshaofeng]

when I install pytorch-lightning==1.5.0, and turn back "pl.Trainer(precision=32, callbacks=[logger])" to "pl.Trainer(gpus=1, precision=32, callbacks=[logger])". It gives the training process like follow:

Is it the right speed?

[tengshaofeng]

pip install pytorch-lightning==2.2.3 and modify the code to "trainer = pl.Trainer(accelerator='gpu', devices=1, precision=32, callbacks=[logger])" is also work.