The error remained.
But I see this now:
input [1], target [32] - if there’s 1 jpg file in the folder
input [2], target [32] - if there’s 2 files
etc
instead of
input [1 x 1], target [32]
input [2 x 1], target [32]
But if I have 32 or more files, I get input [32 x 1], target [32] again
And yes it seems this problem can be connected with squeeze()
Here is something similar. 35