From ad5eec314c0b82005c361249acc4918d36cb0d90 Mon Sep 17 00:00:00 2001
From: root <“junnan.li@salesforce.com”>
Date: Fri, 11 Feb 2022 07:55:17 +0000
Subject: [PATCH] update demo

---
 demo.ipynb | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/demo.ipynb b/demo.ipynb
index 0755de1..62e6c3c 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -106,7 +106,10 @@
     "model = model.to(device)\n",
     "\n",
     "with torch.no_grad():\n",
-    "    caption = model.generate(image, sample=False, num_beams=3, max_length=20, min_length=5)\n",
+    "    # beam search\n",
+    "    caption = model.generate(image, sample=False, num_beams=3, max_length=20, min_length=5) \n",
+    "    # nucleus sampling\n",
+    "    # caption = model.generate(image, sample=True, top_p=0.9, max_length=20, min_length=5) \n",
     "    print('caption: '+caption[0])"
    ]
   },