На этом шаге мы рассмотрим реализацию механизма итерации по пакетам.

Нам также нужен механизм, который поможет выполнять итерации по пакетам и изображениям из набора MNIST. Реализуется такой механизм просто:

import math


class BatchGenerator:
    def __init__(self, images, labels, batch_size=128):
        assert len(images) == len(labels) 
        self.index = 0 
        self.images = images 
        self.labels = labels 
        self.batch_size = batch_size
        self.num_batches = math.ceil(len(images) / batch_size) 

    def next(self):
        images = self.images[self.index:self.index + self.batch_size] 
        labels = self.labels[self.index:self.index + self.batch_size] 
        self.index += self.batch_size 
        return images, labels

На следующем шаге мы рассмотрим выполнение одного этапа обучения.

Предыдущий шаг Содержание Следующий шаг