Could you show us the dataset's format？

[biaozhou]

mnist's format?

[TinyZeaMays]

for img, label in tqdm(train_loader):
print(img, label)

[biaozhou]

I run this project, but it comes with some problems. The problems are the following. Could you give me some advice?

[TinyZeaMays]

I tested on pytorch 1.4.0. Some version may not support .mean(dim=[2, 3]), .squeeze(-1).squeeze(-1) may help.

[biaozhou]

Thank you. I will try it.