1

owenliang · Jun 18, 2023 · ca0acf2 · ca0acf2
1 parent f27cc69
commit ca0acf2
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/decoder.py b/decoder.py
@@ -3,7 +3,6 @@
 '''
 from torch import nn 
 import torch 
-from multihead_attn import MultiHeadAttention
 from emb import EmbeddingWithPosition
 from dataset import de_preprocess,en_preprocess,train_dataset,de_vocab,PAD_IDX,en_vocab
 from decoder_block import DecoderBlock

diff --git a/train.py b/train.py
@@ -3,8 +3,6 @@
 '''
 from torch import nn 
 import torch
-from decoder import Decoder
-from encoder import Encoder
 from dataset import en_preprocess,de_preprocess,train_dataset,en_vocab,de_vocab,PAD_IDX
 from transformer import Transformer
 from torch.utils.data import DataLoader,Dataset