fix a bug in AddCheckpoints

facebookarchive · Jan 15, 2019 · fdd1d4f · fdd1d4f
1 parent c762d55
commit fdd1d4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CIFAR10_Part1.ipynb b/CIFAR10_Part1.ipynb
@@ -622,8 +622,8 @@
     "\n",
     "# Add checkpoints to a given model\n",
     "def AddCheckpoints(model, checkpoint_iters, db_type):\n",
-    "    ITER = brew.iter(train_model, \"iter\")\n",
-    "    train_model.Checkpoint([ITER] + train_model.params, [], db=os.path.join(unique_timestamp, \"cifar10_checkpoint_%05d.lmdb\"), db_type=\"lmdb\", every=checkpoint_iters)"
+    "    ITER = brew.iter(model, \"iter\")\n",
+    "    model.Checkpoint([ITER] + model.params, [], db=os.path.join(unique_timestamp, \"cifar10_checkpoint_%05d.lmdb\"), db_type=\"lmdb\", every=checkpoint_iters)"
    ]
   },
   {