Смена класса в контейнере в зависимости от цикла WP

<div class="<?php echo $i >= 3 ? 'col-md-6' : 'col-md-4'; ?>">

                    <?php $i++; ?>

                    <?php $featured_img_url = get_the_post_thumbnail_url(get_the_ID(),'full');?>
                    <div class="direction-item" style="background-image: url(<?php echo $featured_img_url; ?>)">
                    
                    <?php
                        /* Значение "full" означает, что мы получаем полноразмерную картинку, также можно вставлять другие значение small и medium */
                        $featured_img_url = get_the_post_thumbnail_url(get_the_ID(),'full');
                    ?>
                    
                    
                        <h3><?php the_title(); ?></h3>
                        
                        <div class="direction-desc"><?php the_excerpt(); ?></div>
                        
                        <a href="<?php the_permalink()?>">Подробнее</a>
                    </div>
                </div>