diff --git a/ginka/dataset.py b/ginka/dataset.py
index 512d0a2..9fcf55d 100644
--- a/ginka/dataset.py
+++ b/ginka/dataset.py
@@ -1,12 +1,6 @@
 import json
-import math
-import random
 import torch
-import torch.nn.functional as F
 from torch.utils.data import Dataset
-import torch
-import torch.nn.functional as F
-from typing import List
 
 def load_data(path: str):
     with open(path, 'r', encoding="utf-8") as f:
diff --git a/ginka/maskGIT/mask.py b/ginka/maskGIT/mask.py
index 0cfb31a..72a78ae 100644
--- a/ginka/maskGIT/mask.py
+++ b/ginka/maskGIT/mask.py
@@ -1,5 +1,4 @@
 import random
-import torch
 import numpy as np
 from scipy.ndimage import binary_dilation, binary_erosion
 
diff --git a/ginka/train_maskGIT.py b/ginka/train_maskGIT.py
index 8722fd9..9faa29d 100644
--- a/ginka/train_maskGIT.py
+++ b/ginka/train_maskGIT.py
@@ -9,8 +9,8 @@ import torch.nn.functional as F
 import torch.optim as optim
 import cv2
 import numpy as np
-from torch_geometric.loader import DataLoader
 from tqdm import tqdm
+from torch.utils.data import DataLoader
 from .maskGIT.model import GinkaMaskGIT
 from .dataset import GinkaMaskGITDataset
 from shared.image import matrix_to_image_cv
@@ -61,7 +61,7 @@ disable_tqdm = not sys.stdout.isatty()
 def parse_arguments():
     parser = argparse.ArgumentParser(description="training codes")
     parser.add_argument("--resume", type=bool, default=False)
-    parser.add_argument("--state_ginka", type=str, default="result/vae/ginka-100.pth")
+    parser.add_argument("--state_ginka", type=str, default="result/transformer/ginka-100.pth")
     parser.add_argument("--train", type=str, default="ginka-dataset.json")
     parser.add_argument("--validate", type=str, default="ginka-eval.json")
     parser.add_argument("--epochs", type=int, default=100)
diff --git a/requirements.txt b/requirements.txt
index aa7eed8..c451531 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 torch
-torchvision
-torchaudio
 tqdm
 torch-geometric
 transformers
-torch-scatter
\ No newline at end of file
+scipy
+numpy
+cv2
\ No newline at end of file
diff --git a/train.txt b/train.txt
deleted file mode 100644
index fba325c..0000000
--- a/train.txt
+++ /dev/null
@@ -1 +0,0 @@
-python3 -u -m ginka.train_wgan --epochs 200 --checkpoint 20 --resume true --state_ginka result/wgan/ginka-400.pth --state_minamo result/wgan/minamo-400.pth >> output.log
\ No newline at end of file