К июню 1941 года старший лейтенант А. И. Козьмин служил помощником начальника штаба того же полка в Белорусском Особом военном округе, в Брестской области Белорусской ССР. Участник Великой Отечественной войны с первого дня, вместе с дивизией сражался в 4-й армии Западного фронта. Участвовал в трагическом Белостокско-Минском сражении, попал в окружение. Считался пропавшим без вести, однако сумел вывести из окружения отряд красноармейцев в 90 человек и вынес на своём теле Боевое знамя полка.