Merge pull request falloutdurham#43 from MarcusFra/ch04_range

Change range() to count epochs from 1

Author: falloutdurham

chapter4/Chapter 4.ipynb

@@ -84,7 +84,7 @@
    "outputs": [],
    "source": [
     "def train(model, optimizer, loss_fn, train_loader, val_loader, epochs=20, device=\"cpu\"):\n",
-    "    for epoch in range(epochs):\n",
+    "    for epoch in range(1, epochs+1):\n",
     "        training_loss = 0.0\n",
     "        valid_loss = 0.0\n",
     "        model.train()\n",
@@ -131,11 +131,12 @@
     "        return True\n",
     "    except:\n",
     "        return False\n",
+    "\n",
     "img_transforms = transforms.Compose([\n",
     "    transforms.Resize((64,64)),    \n",
     "    transforms.ToTensor(),\n",
     "    transforms.Normalize(mean=[0.485, 0.456, 0.406],\n",
-    "                    std=[0.229, 0.224, 0.225] )\n",
+    "                         std=[0.229, 0.224, 0.225] )\n",
     "    ])\n",
     "train_data_path = \"./train/\"\n",
     "train_data = torchvision.datasets.ImageFolder(root=train_data_path,transform=img_transforms, is_valid_file=check_image)\n",
@@ -144,6 +145,7 @@
     "batch_size=64\n",
     "train_data_loader = torch.utils.data.DataLoader(train_data, batch_size=batch_size, shuffle=True)\n",
     "val_data_loader  = torch.utils.data.DataLoader(val_data, batch_size=batch_size, shuffle=True)\n",
+    "\n",
     "if torch.cuda.is_available():\n",
     "    device = torch.device(\"cuda\") \n",
     "else:\n",
@@ -175,7 +177,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "train(transfer_model, optimizer,torch.nn.CrossEntropyLoss(), train_data_loader,val_data_loader, epochs=5, device=device)"
+    "train(transfer_model, optimizer,torch.nn.CrossEntropyLoss(), train_data_loader, val_data_loader, epochs=5,\n",
+    "      device=device)"
    ]
   },
   {
@@ -406,4 +409,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}