Updated the moco notebook

KevinMusgrave · Jan 29, 2023 · 20a3903 · 20a3903
1 parent bdd7516
commit 20a3903
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/examples/notebooks/MoCoCIFAR10.ipynb b/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",

diff --git a/src/pytorch_metric_learning/__init__.py b/src/pytorch_metric_learning/__init__.py
@@ -1 +1 @@
-__version__ = "2.0.0.dev0"
+__version__ = "2.0.0.dev1"
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		__version__ = "2.0.0.dev0"
		__version__ = "2.0.0.dev1"