Error when training: 'list' object has no attribute 'shape'

[petercmh01]

Hi: thanks for the awsome implementation. I encounter an error when training on colab that says the batch images are a list instead of tensor (which does not allow all the .shape method in the code).

I'm not sure if that is cause by the dataloader I wrote. I have attached both the error and my dataloader.

Can you help me check it? Thank you in advance : )