Error training PoseC3D on NTU RGB+D dataset

[phemiloyeAI]

I'm working on using PoseC3D on the "NTU RGB+D 120" dataset to recongnize medical condition actions.

I downloaded the already processed data (pickled) from here and I removed the labels I didn't need.

The configuration file I'm using here

First, when I tried to run the training script, I had an error about "img_shape" not stated in the dataset, so I created a key in the each annotation and added image shape as 256x256.

annots['img_shape'] = (256, 256)

That resolved the issue, but now I have another error which I can't make sense of. I'll share the snapshot of the error
this is the error I'm having

[phemiloyeAI]

I used NTU RGB+D 120 2D pickle file instead of the 3D