Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 7bb42712 authored by patrickvonplaten's avatar patrickvonplaten
Browse files

remove ipdb debugging statements

parent 267587c2
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
src/transformers/modeling_utils.py
View file @ 7bb42712
...@@ -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