The Train Deep Learning Model is failing on me in ArcGIS Pro 3.4.2 even though I have even number of training features.
I am btw also using a pre-trained model (Water Body Extraction (SAR) - USA) as well in the tool.
Validation set at 10
Here's a snippet of stat file
Tool
Error
@PavanYadav any clues?
@Ed_ looks like you have 6 image chips in the training data..
i do have 600 training features, so what do you mean by 50-100 samples? Are those the number of image chips? I do have Nvidia RTX A2000 12 gig, so should i still decrease the tile size?
@Ed_ you can try 2, but to be honest; I would have at least 50-100 samples just to be on the safer side.. I see your tile size is 512, which could be fairly large for your GPU.. maybe decease it to 256 and use a stride of 128.. these will produce a lot more image chips..
Hi Pavan, happy Friday, thank you for the response, so in this case what should be batch size then?
hi @Ed_ we need to handle this error bit better and provide you with actionable information. One reason that I have seen this occur is that when the batch size is larger than the number of chips in the data, the Train Deep Learning Model tool may return "ZeroDivisionError: division by zero" in this case.
It looks like you have 5 images chips and your batch size if 8.
I am on the latest version of Pro that is 3.4.2. I am surprised that it's giving me the zero division error, because as you can see in the above stat file screenshot, the training data has 600 features.
Hi Shivani, thank you for the reply, yes I basically created a very small training subset from the USA Wetlands data available on the Living Atlas. In that subset I then added an attribute for class value of 1.
Also which ArcGIS Pro version are you using?
Hi @Ed_ , could you please confirm if you've referred to the guide for creating the training data?
@BrendanNewell @ShivaniPathak can you please help with this? Thank you!
Signed in members can post, follow updates, and more. New here? Register a free account.
Find useful guides, FAQs, and documents to help you navigate and make the most of Esri Community.
