Skip to content

Instantly share code, notes, and snippets.

View kushalchauhan98's full-sized avatar
🔥
Setting TPUs on fire!

Kushal Chauhan kushalchauhan98

🔥
Setting TPUs on fire!
View GitHub Profile
@kushalchauhan98
kushalchauhan98 / pad_packed_demo.py
Created May 20, 2019 12:33 — forked from Tushar-N/pad_packed_demo.py
How to use pad_packed_sequence in pytorch
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