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

Fix flake8

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