b03902130

Tushar-N / pad_packed_demo.py

Last active October 27, 2024 15:17

How to use pad_packed_sequence in pytorch<1.1.0

	import torch
	import torch.nn as nn
	from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence

	seqs = ['gigantic_string','tiny_str','medium_str']

	# make <pad> idx 0
	vocab = ['<pad>'] + sorted(set(''.join(seqs)))

	# make model