remove ipdb debugging statements

This commit is contained in:
patrickvonplaten committed 2019-12-25 23:17:24 +01:00
1 parent 267587c258
commit 7bb4271291
1 file changed
-2
-2
View File
@@ -18,7 +18,6 @@
import logging
import os
import ipdb
import torch
from torch import nn
@@ -794,7 +793,6 @@ class PreTrainedModel(nn.Module):
):
""" Generate sequences for each example with beam search.
"""
ipdb.set_trace()
# Expand input to num beams
input_ids = input_ids.unsqueeze(1).expand(batch_size, num_beams, cur_len)
input_ids = input_ids.contiguous().view(batch_size * num_beams, cur_len) # (batch_size * num_beams, cur_len)