diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..4a36a8d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,7 @@
+utils/__pycache__/*
+outputs/*
+models/MSG3D/__pycache__/*
+logs/*
+models/__pycache__/*
+data/*
+*test*
\ No newline at end of file
diff --git a/config/SiT/training/hparams.yml b/config/SiT/training/hparams.yml
index 8f11633..96c8c77 100644
--- a/config/SiT/training/hparams.yml
+++ b/config/SiT/training/hparams.yml
@@ -17,7 +17,7 @@ training:
   LR: 0.00001
   bs: 256
   bs_val: 1
-  epochs: 30
+  epochs: 100
   gpu: 0
   l1loss: False
   testing: False
@@ -78,7 +78,4 @@ sub_ico_1:
 
 sub_ico_2:
     num_patches: 320
-    num_vertices: 153 
-
-
-
+    num_vertices: 153 
\ No newline at end of file
diff --git a/models/sit.py b/models/sit.py
index ed24def..e2a5137 100644
--- a/models/sit.py
+++ b/models/sit.py
@@ -20,7 +20,7 @@
 from einops import repeat
 from einops.layers.torch import Rearrange
 
-from vit_pytorch.vit import Transformer, Attention, FeedForward, PreNorm
+from vit_pytorch.vit import Transformer
 
 class SiT(nn.Module):
     def __init__(self, *,
diff --git a/tools/train.py b/tools/train.py
index a5376cd..b1c820a 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -39,14 +39,8 @@
 import numpy as np
 import pandas as pd
 
-from torch.optim.lr_scheduler import StepLR
-from torch.optim.lr_scheduler import ReduceLROnPlateau
-
-
 from models.sit import SiT
 
-from warmup_scheduler import GradualWarmupScheduler
-
 from utils.utils import load_weights_imagenet
 
 from torch.utils.tensorboard import SummaryWriter