Commit 7bb42712 authored by patrickvonplaten's avatar patrickvonplaten
Browse files

remove ipdb debugging statements

parent 267587c2
...@@ -18,7 +18,6 @@ ...@@ -18,7 +18,6 @@
import logging import logging
import os import os
import ipdb
import torch import torch
from torch import nn from torch import nn
...@@ -794,7 +793,6 @@ class PreTrainedModel(nn.Module): ...@@ -794,7 +793,6 @@ class PreTrainedModel(nn.Module):
): ):
""" Generate sequences for each example with beam search. """ Generate sequences for each example with beam search.
""" """
ipdb.set_trace()
# Expand input to num beams # Expand input to num beams
input_ids = input_ids.unsqueeze(1).expand(batch_size, num_beams, cur_len) 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) input_ids = input_ids.contiguous().view(batch_size * num_beams, cur_len) # (batch_size * num_beams, cur_len)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment