revision

Author: dvgodoy

Chapter01.ipynb

@@ -123,7 +123,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "OrderedDict([('0.weight', tensor([[1.9690]], device='cuda:0')), ('0.bias', tensor([1.0235], device='cuda:0'))])\n"
+      "OrderedDict([('0.weight', tensor([[1.9690]])), ('0.bias', tensor([1.0235]))])\n"
      ]
     }
    ],
@@ -327,7 +327,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Writing model_configuration/v1.py\n"
+      "Overwriting model_configuration/v1.py\n"
      ]
     }
    ],
@@ -370,7 +370,7 @@
     {
      "data": {
       "text/plain": [
-       "<function __main__.make_train_step.<locals>.perform_train_step(x, y)>"
+       "<function __main__.make_train_step_fn.<locals>.perform_train_step_fn(x, y)>"
       ]
      },
      "execution_count": 18,
@@ -398,7 +398,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Writing model_training/v1.py\n"
+      "Overwriting model_training/v1.py\n"
      ]
     }
    ],
@@ -435,7 +435,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "OrderedDict([('0.weight', tensor([[1.9690]], device='cuda:0')), ('0.bias', tensor([1.0235], device='cuda:0'))])\n"
+      "OrderedDict([('0.weight', tensor([[1.9690]])), ('0.bias', tensor([1.0235]))])\n"
      ]
     }
    ],
@@ -556,37 +556,37 @@
     {
      "data": {
       "text/plain": [
-       "[tensor([[0.1196],\n",
-       "         [0.1395],\n",
-       "         [0.2809],\n",
-       "         [0.1834],\n",
-       "         [0.3585],\n",
+       "[tensor([[0.9507],\n",
        "         [0.5427],\n",
-       "         [0.0885],\n",
-       "         [0.9489],\n",
-       "         [0.9699],\n",
-       "         [0.7751],\n",
-       "         [0.9696],\n",
-       "         [0.7320],\n",
-       "         [0.0055],\n",
-       "         [0.7069],\n",
-       "         [0.8155],\n",
-       "         [0.5979]]), tensor([[1.3214],\n",
-       "         [1.3051],\n",
-       "         [1.5846],\n",
-       "         [1.4637],\n",
-       "         [1.7462],\n",
+       "         [0.1409],\n",
+       "         [0.3745],\n",
+       "         [0.1987],\n",
+       "         [0.8948],\n",
+       "         [0.7722],\n",
+       "         [0.7852],\n",
+       "         [0.5248],\n",
+       "         [0.2809],\n",
+       "         [0.1159],\n",
+       "         [0.0740],\n",
+       "         [0.1849],\n",
+       "         [0.4561],\n",
+       "         [0.7608],\n",
+       "         [0.1560]]), tensor([[2.8715],\n",
        "         [2.2161],\n",
-       "         [1.0708],\n",
-       "         [2.8903],\n",
-       "         [2.9727],\n",
-       "         [2.4936],\n",
-       "         [2.8401],\n",
-       "         [2.4732],\n",
-       "         [1.0632],\n",
-       "         [2.4388],\n",
-       "         [2.6606],\n",
-       "         [2.0407]])]"
+       "         [1.1211],\n",
+       "         [1.7578],\n",
+       "         [1.2654],\n",
+       "         [2.7393],\n",
+       "         [2.4208],\n",
+       "         [2.5283],\n",
+       "         [2.0167],\n",
+       "         [1.5846],\n",
+       "         [1.1603],\n",
+       "         [1.1713],\n",
+       "         [1.5888],\n",
+       "         [1.7706],\n",
+       "         [2.4970],\n",
+       "         [1.2901]])]"
       ]
      },
      "execution_count": 25,
@@ -614,7 +614,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Writing data_preparation/v1.py\n"
+      "Overwriting data_preparation/v1.py\n"
      ]
     }
    ],
@@ -666,7 +666,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Writing model_training/v2.py\n"
+      "Overwriting model_training/v2.py\n"
      ]
     }
    ],
@@ -718,7 +718,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "OrderedDict([('0.weight', tensor([[1.9684]], device='cuda:0')), ('0.bias', tensor([1.0235], device='cuda:0'))])\n"
+      "OrderedDict([('0.weight', tensor([[1.9694]])), ('0.bias', tensor([1.0234]))])\n"
      ]
     }
    ],
@@ -1013,7 +1013,7 @@
     "\n",
     "for epoch in range(n_epochs):\n",
     "    # inner loop\n",
-    "    loss = mini_batch(device, train_loader, train_step)\n",
+    "    loss = mini_batch(device, train_loader, train_step_fn)\n",
     "    losses.append(loss)\n",
     "    \n",
     "    # VALIDATION\n",
@@ -1267,7 +1267,7 @@
     "loss_fn = nn.MSELoss(reduction='mean')\n",
     "\n",
     "# Creates the train_step function for our model, loss function and optimizer\n",
-    "train_step = make_train_step(model, loss_fn, optimizer)\n",
+    "train_step_fn = make_train_step_fn(model, loss_fn, optimizer)\n",
     "\n",
     "# Creates the val_step function for our model and loss function\n",
     "val_step_fn = make_val_step_fn(model, loss_fn)\n",
@@ -1320,7 +1320,7 @@
     "\n",
     "for epoch in range(n_epochs):\n",
     "    # inner loop\n",
-    "    loss = mini_batch(device, train_loader, train_step)\n",
+    "    loss = mini_batch(device, train_loader, train_step_fn)\n",
     "    losses.append(loss)\n",
     "    \n",
     "    # VALIDATION\n",
@@ -1812,7 +1812,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.7.5"
   }
  },
  "nbformat": 4,