+ results = autoregression(
+ model, self.batch_size,
+ nb_samples = 1, nb_tokens = nb_tokens,
+ primer = self.tensorize(primer_descr),
+ device = self.device
+ )
+ r = ' '.join([ self.id2token[t.item()] for t in results.flatten() ])
+ result_descr.append(r)
+
+ img = [
+ picoclvr.descr2img(d, height = self.height, width = self.width)
+ for d in result_descr
+ ]