Hy @Bisma_Akram



 Bisma_Akram:

damage_labels.append(damage_dict[label])
KeyError: ‘D0w0


This happens inside your dataloader Because you don’t have a specific key for that.



 Bisma_Akram:

damage_dict={

        'D00': 1,

        'D01': 2, 

        'D10': 3,

        'D11': 4,

        'D20': 5,

        'D40': 6,

        'D43': 7,

        'D44': 8,

        'D50': 9




The key generating error is this



 Bisma_Akram:

KeyError: ‘D0w0’