diff --git a/scripts/Ours/stage2_pretrain_encoder.py b/scripts/Ours/stage2_pretrain_encoder.py
index 150014e..1b0e0e5 100644
--- a/scripts/Ours/stage2_pretrain_encoder.py
+++ b/scripts/Ours/stage2_pretrain_encoder.py
@@ -25,9 +25,8 @@
 from robo_utils.oxford.oxford_dataset import DIVADataset
 from robo_utils.kitti.torch_dataset import OurDataset as KittiDataset
 from learning.dataset import CARLADataset
-from utils import write_params, check_shape, to_device, set_mute
+from utils import write_params, to_device
 from carla_utils import parse_yaml_file_unsafe
-from torchvision.utils import save_image
 
 random.seed(datetime.now())
 torch.manual_seed(666)
@@ -76,8 +75,6 @@
 trajectory_criterion = torch.nn.MSELoss().to(device)
 
 e_optimizer = torch.optim.Adam(encoder.parameters(), lr=opt.lr, weight_decay=opt.weight_decay)
-e_optimizer_woirm = torch.optim.Adam(encoder_woirm.parameters(), lr=opt.lr, weight_decay=opt.weight_decay)
-
 
 param = parse_yaml_file_unsafe('./param_oxford.yaml')
 train_loader = DataLoader(DIVADataset(param, mode='train', opt=opt), batch_size=opt.batch_size, shuffle=False, num_workers=opt.n_cpu)
diff --git a/scripts/Ours/stage3_train_e2e.py b/scripts/Ours/stage3_train_e2e.py
index 95758fb..60b9b5e 100644
--- a/scripts/Ours/stage3_train_e2e.py
+++ b/scripts/Ours/stage3_train_e2e.py
@@ -22,7 +22,6 @@
 from learning.dataset import CARLADataset
 from utils import write_params, to_device
 from carla_utils import parse_yaml_file_unsafe
-from torchvision.utils import save_image
 
 random.seed(datetime.now())
 torch.manual_seed(666)