AnsweredAssumed Answered

Deep learning - invalid argument 0: Sizes of tensors must match except in dimension 0

Question asked by vholubecarcdata-cz-esridist Employee on Nov 29, 2019
Latest reply on Dec 4, 2019 by s.kumar-esristaff

Dear all,

 

I have issue with deep learning tools. I exported 12 000 training chips from Esri Imagery basemap to TIFF - see settings.png

 

The TIFFs looks fine as well as the labels file.

 

On calling data = prepare_data(data_path, class_mapping={1:'Pool'}, batch_size=64) I am getting error - UserWarning: It's not possible to collate samples of your dataset together in a batch.
warn(message)

 

For data.show_batch() -  now I am getting result - RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 19 and 10 in dimension 3 at ..\aten\src\TH/generic/THTensor.cpp:711

I am not sure what does it mean? Does the Export training data tool corrupt some chips? Is there a way how to find out which chips are corrupted? Or how to fix it? 

 

The chips were created in PRO by Export training data tool, so these could be fine.

 

Thank you for any idea.

 

Vladimir

Attachments

Outcomes