Commit 5935fe2f authored by Myle Ott's avatar Myle Ott
Browse files

Fix flake8

parent f68a4435
......@@ -7,7 +7,6 @@
import math
import torch
from torch.autograd import Variable
import torch.nn as nn
import torch.nn.functional as F
......
......@@ -26,9 +26,6 @@ class SinusoidalPositionalEmbedding(nn.Module):
self.embedding_dim = embedding_dim
self.padding_idx = padding_idx
self.left_pad = left_pad
self.register_buffer('range_buf', None)
self._cache = {}
self.register_buffer(
'weights',
SinusoidalPositionalEmbedding.get_embedding(
......
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