В моем предыдущем посте (перейти по ссылке) я рассказал о построении вашей нейронной сети с использованием модуля nn, предлагаемого PyTorch.
В этой статье я собираюсь показать вам, как вы можете создать такую же нейронную сеть, используя модуль Sequential в PyTorch. Но перед этим
что такое модуль Sequential?
nn.Sequential – это модуль, который содержит другие модули и последовательно применяет их для получения вывода. Каждый линейный модуль вычисляет выходные данные из входных данных, используя линейную функцию, и содержит внутренние тензоры для своего веса и смещения. (ссылка: ссылка)
Итак, теперь давайте начнем:
Шаг 1. Подготовьте данные (цель — научить модель вычитанию)
Шаг 2. Нам нужен Dataset и DataLoader для пакетной обработки наших данных
Вы можете использовать iter и next для просмотра элементов в загрузчике данных следующим образом:
Шаг 3. Создайте нейронную сеть
Шаг 4. Обучение модели
Шаг 5. Нанесите на график потери с учетом эпох
Шаг 6. Моделирование на основе неизвестных данных
Шаг 7. Распечатайте сводку модели
Хорошо, это было легко! :)
Оставьте комментарий с любыми предложениями. Спасибо, что нашли время, чтобы прочитать эту статью.
*****************************Конец******************* ********