Training speed.

[andreYoo]

I am using two A6000 RTX GPUs.

When I train the autoencoder models with the ImageNet dataset.

it shows 2.14s/it.

is it normal?

[hahamidi]

As I remember, "s/it" stands for seconds per batch, which greatly depends on your batch size. With a standard batch size, yes, this duration is typical. Additionally, it's highly dependent on the number of workers you have in your dataloader.