Merge pull request gan3sh500#4 from fperdigon/master

Fixing stupid mistakes.

Author: gan3sh500

notebook.ipynb

@@ -199,7 +199,7 @@
         "def mixmatch(x, y, u, model, augment_fn, T=0.5, K=2, alpha=0.75):\n",
         "    xb = augment_fn(x)\n",
         "    ub = [augment_fn(u) for _ in range(K)]\n",
-        "    qb = sharpen(sum(map(lambda i: model(i), ub)) / K)\n",
+        "    qb = sharpen(sum(map(lambda i: model(i), ub)) / K, T)\n",
         "    Ux = np.concatenate(ub, axis=0)\n",
         "    Uy = np.concatenate([qb for _ in range(K)], axis=0)\n",
         "    indices = np.random.shuffle(np.arange(len(xb) + len(Ux)))\n",
@@ -237,7 +237,7 @@
         "        self.mse = torch.nn.MSELoss()\n",
         "        super(MixMatchLoss, self).__init__()\n",
         "    \n",
-        "    def forward(X, U, p, q, model):\n",
+        "    def forward(self, X, U, p, q, model):\n",
         "        X_ = np.concatenate([X, U], axis=1)\n",
         "        preds = model(X_)\n",
         "        return self.xent(preds[:len(p)], p) + \\\n",
@@ -482,4 +482,4 @@
       "outputs": []
     }
   ]
-}
+}