Update training graph