APE and AVE_pose metric calculation

Hey, Thank you so much for sharing this project.
I was trying to run the training code and saw what I think is a little bug here

TEMOS/temos/model/base.py

Line 45 in ea12cf6

    
           dico.update({f"Metrics/{metric}": value for metric, value in metrics_dict.items()})

Should it have a .mean()?
dico.update({f"Metrics/{metric}": value.mean() for metric, value in metrics_dict.items()})
Otherwise, I get this error

Please let me know if I'm right or not :)

Thanks!

Hello @angelacast135,

Thanks for the pointer, I just got the error myself when upgrading the packages.
I just fixed this issue by removing the logging of the arrays (APE_pose, APE_joints, AVE_pose, AVE_joints) as it is not necessary (and now not supported apparently). I am already logging the means. It solves the problem.

Commit: 40410d9