Hi,
How many samples are there in your batch initially? 1024 or 64?
If it’s 1024 then your target size if wrong and you should find out why you only have 64 targets.
If it’s 64, then you need to find out why x is of size 1024 while it should only be 64.