立装甲兵训练中心，仍担任装甲兵司令。1952年，大整编，2个装甲总队被改编为装甲第1、2师，按美军1945年型装甲师进行编组，辖A、B、预备（R）3个战斗群。完成装甲部队整编工作，与于豪章编定装甲部队作战准则。1953年两度赴美，先是随徐培根赴美考察访问，回国时元配石静宜去世，蒋介石怕蒋纬国在台北触景伤情，所以送蒋纬国入美国陆军指挥参谋学院正规班及防空学校飞弹班受训。美国受训完后调国防部第三厅副厅长，转任国防部第五厅厅长。1958年9月复回任装甲兵司令。1959年增编装甲部队，增设装骑连编制