{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "from utils import Scalar, draw_graph\n", "from linear_model import Linear, mse" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [ { "data": { "image/svg+xml": [ "\n", "\n", "\n", "\n", "\n", "\n", "%3\n", "\n", "\n", "140456118688784backward\n", "\n", "grad= 4.50\n", "\n", "value= 4.50\n", "\n", "-\n", "\n", "\n", "140456118688112backward\n", "\n", "y2= 4.50\n", "\n", "\n", "140456118688784backward->140456118688112backward\n", "\n", "\n", "\n", "140456118688736backward\n", "\n", "grad=-4.50\n", "\n", "value= 0.00\n", "\n", "+\n", "\n", "\n", "140456118688784backward->140456118688736backward\n", "\n", "\n", "-4.50\n", "\n", "\n", "140456118688304backward\n", "\n", "grad=-1.00\n", "\n", "value= 0.00\n", "\n", "*\n", "\n", "\n", "140456118688352backward\n", "\n", "grad=-10.50\n", "\n", "value= 0.00\n", "\n", "a\n", "\n", "\n", "140456118688304backward->140456118688352backward\n", "\n", "\n", "-1.50\n", "\n", "\n", "140456118688160backward\n", "\n", "x1= 1.50\n", "\n", "\n", "140456118688304backward->140456118688160backward\n", "\n", "\n", "\n", "140456118688832backward\n", "\n", "grad= 1.00\n", "\n", "value= 10.62\n", "\n", "mse\n", "\n", "\n", "140456118688832backward->140456118688784backward\n", "\n", "\n", " 4.50\n", "\n", "\n", "140456118688400backward\n", "\n", "grad= 1.00\n", "\n", "value= 1.00\n", "\n", "-\n", "\n", "\n", "140456118688832backward->140456118688400backward\n", "\n", "\n", " 1.00\n", "\n", "\n", "140456118688544backward\n", "\n", "grad=-1.00\n", "\n", "value= 0.00\n", "\n", "+\n", "\n", "\n", "140456118688400backward->140456118688544backward\n", "\n", "\n", "-1.00\n", "\n", "\n", "140456118688064backward\n", "\n", "y1= 1.00\n", "\n", "\n", "140456118688400backward->140456118688064backward\n", "\n", "\n", "\n", "140456118688496backward\n", "\n", "grad=-5.50\n", "\n", "value= 0.00\n", "\n", "b\n", "\n", "\n", "140456118688544backward->140456118688304backward\n", "\n", "\n", "-1.00\n", "\n", "\n", "140456118688544backward->140456118688496backward\n", "\n", "\n", "-1.00\n", "\n", "\n", "140456118688688backward\n", "\n", "grad=-4.50\n", "\n", "value= 0.00\n", "\n", "*\n", "\n", "\n", "140456118688688backward->140456118688352backward\n", "\n", "\n", "-9.00\n", "\n", "\n", "140456118688208backward\n", "\n", "x2= 2.00\n", "\n", "\n", "140456118688688backward->140456118688208backward\n", "\n", "\n", "\n", "140456118688736backward->140456118688496backward\n", "\n", "\n", "-4.50\n", "\n", "\n", "140456118688736backward->140456118688688backward\n", "\n", "\n", "-4.50\n", "\n", "\n", "\n" ], "text/plain": [ "" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# 定义训练数据\n", "x1 = Scalar(1.5, label='x1', requires_grad=False)\n", "y1 = Scalar(1.0, label='y1', requires_grad=False)\n", "x2 = Scalar(2.0, label='x2', requires_grad=False)\n", "y2 = Scalar(4.5, label='y2', requires_grad=False)\n", "# 反向传播\n", "model = Linear()\n", "loss = mse([model.error(x1, y1), model.error(x2, y2)])\n", "loss.backward()\n", "draw_graph(loss, 'backward')" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "data": { "image/svg+xml": [ "\n", "\n", "\n", "\n", "\n", "\n", "%3\n", "\n", "\n", "140456118688256backward\n", "\n", "grad=-1.00\n", "\n", "value= 0.00\n", "\n", "*\n", "\n", "\n", "140456118686384backward\n", "\n", "a= 0.00\n", "\n", "\n", "140456118688256backward->140456118686384backward\n", "\n", "\n", "\n", "140456118688160backward\n", "\n", "x1= 1.50\n", "\n", "\n", "140456118688256backward->140456118688160backward\n", "\n", "\n", "\n", "140456118687248backward\n", "\n", "grad= 1.00\n", "\n", "value= 10.62\n", "\n", "mse\n", "\n", "\n", "140456118688448backward\n", "\n", "grad= 1.00\n", "\n", "value= 1.00\n", "\n", "-\n", "\n", "\n", "140456118687248backward->140456118688448backward\n", "\n", "\n", " 1.00\n", "\n", "\n", "140456118687200backward\n", "\n", "grad= 4.50\n", "\n", "value= 4.50\n", "\n", "-\n", "\n", "\n", "140456118687248backward->140456118687200backward\n", "\n", "\n", " 4.50\n", "\n", "\n", "140456118687392backward\n", "\n", "grad=-1.00\n", "\n", "value= 0.00\n", "\n", "+\n", "\n", "\n", "140456118687392backward->140456118688256backward\n", "\n", "\n", "\n", "140456118687056backward\n", "\n", "grad=-5.50\n", "\n", "value= 0.00\n", "\n", "b\n", "\n", "\n", "140456118687392backward->140456118687056backward\n", "\n", "\n", "-1.00\n", "\n", "\n", "140456118688448backward->140456118687392backward\n", "\n", "\n", "-1.00\n", "\n", "\n", "140456118688064backward\n", "\n", "y1= 1.00\n", "\n", "\n", "140456118688448backward->140456118688064backward\n", "\n", "\n", "\n", "140456118688112backward\n", "\n", "y2= 4.50\n", "\n", "\n", "140456118687104backward\n", "\n", "grad=-4.50\n", "\n", "value= 0.00\n", "\n", "*\n", "\n", "\n", "140456118687104backward->140456118686384backward\n", "\n", "\n", "\n", "140456118688208backward\n", "\n", "x2= 2.00\n", "\n", "\n", "140456118687104backward->140456118688208backward\n", "\n", "\n", "\n", "140456118686144backward\n", "\n", "grad=-4.50\n", "\n", "value= 0.00\n", "\n", "+\n", "\n", "\n", "140456118686144backward->140456118687056backward\n", "\n", "\n", "-4.50\n", "\n", "\n", "140456118686144backward->140456118687104backward\n", "\n", "\n", "\n", "140456118687200backward->140456118688112backward\n", "\n", "\n", "\n", "140456118687200backward->140456118686144backward\n", "\n", "\n", "-4.50\n", "\n", "\n", "\n" ], "text/plain": [ "" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# 反向传播\n", "model = Linear()\n", "# 冻结参数a\n", "model.a.requires_grad = False\n", "loss = mse([model.error(x1, y1), model.error(x2, y2)])\n", "loss.backward()\n", "draw_graph(loss, 'backward')" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [ { "data": { "image/svg+xml": [ "\n", "\n", "\n", "\n", "\n", "\n", "%3\n", "\n", "\n", "140456117343232backward\n", "\n", "grad=-1.00\n", "\n", "value= 0.00\n", "\n", "+\n", "\n", "\n", "140456117343568backward\n", "\n", "b= 0.00\n", "\n", "\n", "140456117343232backward->140456117343568backward\n", "\n", "\n", "\n", "140456118619488backward\n", "\n", "grad=-1.00\n", "\n", "value= 0.00\n", "\n", "*\n", "\n", "\n", "140456117343232backward->140456118619488backward\n", "\n", "\n", "\n", "140456117342272backward\n", "\n", "grad= 1.00\n", "\n", "value= 1.00\n", "\n", "-\n", "\n", "\n", "140456117342272backward->140456117343232backward\n", "\n", "\n", "\n", "140456118688064backward\n", "\n", "y1= 1.00\n", "\n", "\n", "140456117342272backward->140456118688064backward\n", "\n", "\n", "\n", "140456118837312backward\n", "\n", "grad= 4.50\n", "\n", "value= 4.50\n", "\n", "-\n", "\n", "\n", "140456118688112backward\n", "\n", "y2= 4.50\n", "\n", "\n", "140456118837312backward->140456118688112backward\n", "\n", "\n", "\n", "140456117343184backward\n", "\n", "grad=-4.50\n", "\n", "value= 0.00\n", "\n", "+\n", "\n", "\n", "140456118837312backward->140456117343184backward\n", "\n", "\n", "\n", "140456118837360backward\n", "\n", "grad= 1.00\n", "\n", "value= 10.62\n", "\n", "mse\n", "\n", "\n", "140456118837360backward->140456117342272backward\n", "\n", "\n", "\n", "140456118837360backward->140456118837312backward\n", "\n", "\n", "\n", "140456117343376backward\n", "\n", "grad=-4.50\n", "\n", "value= 0.00\n", "\n", "*\n", "\n", "\n", "140456117343520backward\n", "\n", "a= 0.00\n", "\n", "\n", "140456117343376backward->140456117343520backward\n", "\n", "\n", "\n", "140456118688208backward\n", "\n", "x2= 2.00\n", "\n", "\n", "140456117343376backward->140456118688208backward\n", "\n", "\n", "\n", "140456118619488backward->140456117343520backward\n", "\n", "\n", "\n", "140456118688160backward\n", "\n", "x1= 1.50\n", "\n", "\n", "140456118619488backward->140456118688160backward\n", "\n", "\n", "\n", "140456117343184backward->140456117343376backward\n", "\n", "\n", "\n", "140456117343184backward->140456117343568backward\n", "\n", "\n", "\n", "\n" ], "text/plain": [ "" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# 反向传播\n", "model = Linear()\n", "# 冻结参数a和参数b\n", "model.a.requires_grad = False\n", "model.b.requires_grad = False\n", "loss = mse([model.error(x1, y1), model.error(x2, y2)])\n", "loss.backward()\n", "draw_graph(loss, 'backward')" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.8.5" } }, "nbformat": 4, "nbformat_minor": 4 }