Si un assuré qui a recouvré une capacité de gain telle qu'elle est définie à l'article 54 quinquies, paragraphe 2, ci-dessus, fait l'objet d'un traitement, suit des cours ou effectue un stage en vue de sa rééducation professionnelle, la caisse peut, au