yjchun

# from https://discuss.pytorch.org/t/how-to-debug-causes-of-gpu-memory-leaks/6741/19
# collect tensors and could calculate memory size
# like following

import torch
import gc
from functools import reduce
import operator as op

yjchun

import math
import numpy
import torch
import torch.nn as nn
import torch.nn.functional as F
import itertools


class BeamSearch(object):
	def __init__(self, model, henc):