run_training.py joint joint_stage2 not working

[TOMMYWHY]

Hi there,

Thank you for providing the code.

However, when I tried to reproduce it, I encountered an issue. In the joint_stage2, I found that you used steepest_descent_resnet50 with backbone_type='vitae'. But in the definition of steepest_descent_resnet50, there is no code related to vitae.
The screenshots are attached.

I look forward to your reply.

Cheers