arcgis.learn python api IndexError: index 0 is out of bounds for axis 0 with size 0

[dnm]

Hello,

I'm trying to retrain Arcgis's Door and Window model with the ArcGIS Python API, but I'm getting an error. I am new to the field of deep learning. I will share the full code and error below. I tried a lot but I couldn't find how to fix the error. Thank you.

This is the error I got.

 

I am also sharing the full code.

[DanPatterson · Accepted solution]

Did you address the warning in In 4 ?  You seem to be getting an empty set/list

[dnm]

Hello,

Thank you very much for your reply. Actually, I've considered it before. But it was giving the error I mentioned without the error you mentioned. Anyway, based on your answer, I made "val_split_pct= 0.2" and instead of getting an index error, I got a "RuntimeError: CUDA out of memory" error in the image below. I guess I need to make adjustments with the parameters.