Training framework diagram 2