Post-training Module