Fix odd import

This commit is contained in:
Matthias
2023-05-21 09:50:59 +02:00
parent 914195acf4
commit a87b215d67

View File

@@ -1,7 +1,7 @@
import math
import torch
import torch.nn as nn
from torch import nn
"""
@@ -68,7 +68,7 @@ class PyTorchTransformerModel(nn.Module):
return x
class PositionalEncoding(torch.nn.Module):
class PositionalEncoding(nn.Module):
def __init__(self, d_model, max_len=5000):
"""
Args