diff --git a/codes/distill_torchscript.py b/codes/distill_torchscript.py
index 8a4ea1cc..20939a91 100644
--- a/codes/distill_torchscript.py
+++ b/codes/distill_torchscript.py
@@ -2,8 +2,12 @@ import argparse
 import options.options as option
 from models.networks import define_G
 import torch
+import torchvision
+import torch.nn.functional as F
+
 
 if __name__ == "__main__":
+
     parser = argparse.ArgumentParser()
     parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/use_vrp_upsample.yml')
     opt = option.parse(parser.parse_args().opt, is_train=False)
diff --git a/codes/test.py b/codes/test.py
index edc70045..5f17b180 100644
--- a/codes/test.py
+++ b/codes/test.py
@@ -17,7 +17,7 @@ if __name__ == "__main__":
     torch.backends.cudnn.benchmark = True
     want_just_images = True
     parser = argparse.ArgumentParser()
-    parser.add_argument('-opt', type=str, help='Path to options YMAL file.', default='../options/use_vrp_upsample.yml')
+    parser.add_argument('-opt', type=str, help='Path to options YMAL file.', default='../options/test_ESRGAN_adrianna_full.yml')
     opt = option.parse(parser.parse_args().opt, is_train=False)
     opt = option.dict_to_nonedict(opt)
 
diff --git a/codes/train.py b/codes/train.py
index 9a19a558..6a09b391 100644
--- a/codes/train.py
+++ b/codes/train.py
@@ -30,7 +30,7 @@ def init_dist(backend='nccl', **kwargs):
 def main():
     #### options
     parser = argparse.ArgumentParser()
-    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_vix_rrdb_v2.yml')
+    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/finetune_vix_resgenv2.yml')
     parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none',
                         help='job launcher')
     parser.add_argument('--local_rank', type=int, default=0)