Updated the moco notebook

Author: KevinMusgrave

examples/notebooks/MoCoCIFAR10.ipynb

@@ -478,7 +478,8 @@
     "    # add an offset so that the labels do not overlap with any labels in the memory queue\n",
     "    labels += previous_max_label + 1\n",
     "    # we want to enqueue the output of encK, which is the 2nd half of the batch\n",
-    "    enqueue_mask = torch.arange(num_pos_pairs, num_pos_pairs * 2)\n",
+    "    enqueue_mask = torch.zeros(len(labels)).bool()\n",
+    "    enqueue_mask[num_pos_pairs:] = True\n",
     "    return labels, enqueue_mask\n",
     "\n",
     "\n",

src/pytorch_metric_learning/__init__.py

@@ -1 +1 @@
-__version__ = "2.0.0.dev0"
+__version__ = "2.0.0.dev1"