1: $

2:     \State Check convergence of joint training

3: \EndFor

4: \For{$