From a8bc9a3b859b4efd77800d3424174e7c8171fd71 Mon Sep 17 00:00:00 2001 From: sayantanbasu05 Date: Sun, 23 Aug 2020 15:27:00 +0530 Subject: [PATCH] Some updates to generate.py --- generate.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/generate.py b/generate.py index ec31c42..733fe30 100644 --- a/generate.py +++ b/generate.py @@ -1,6 +1,4 @@ -""" -@uthor: Prakhar -""" + import os import argparse import torch @@ -23,7 +21,7 @@ def choose_from_top_k_top_n(probs, k=50, p=0.8): f.append(k) pr.append(v) if t>=p: - break + break top_prob = pr / np.sum(pr) token_id = np.random.choice(f, 1, p = top_prob)