train in coco dataset

[guoees]

Hello, I am using coco dataset to train, after 19 epochs of training I get this error

[Arthur151]

It seems that this error occurs during visulization via pyrender.

You can try to debug the error via replacing this line from rendered_img = self.renderer(verts, faces, colors=color, focal_length=args().focal_length, cam_params=cam_params) to

try:
    rendered_img = self.renderer(verts, faces, colors=color, focal_length=args().focal_length, cam_params=cam_params)
except:
    print('vertices shape during visualization:',verts.shape,faces.shape)
    rendered_img = np.zeros((len(verts), 512,512,3)).astype(np.uint8)

I guess there is something wrong with verts shape.