diff --git a/.idea/misc.xml b/.idea/misc.xml
index d9f696a7..966d4b11 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -3,5 +3,5 @@
   <component name="JavaScriptSettings">
     <option name="languageLevel" value="ES6" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (pytorch-1.7)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="torch-cu11" project-jdk-type="Python SDK" />
 </project>
\ No newline at end of file
diff --git a/.idea/mmsr.iml b/.idea/mmsr.iml
index 35f8a980..2d68ad0f 100644
--- a/.idea/mmsr.iml
+++ b/.idea/mmsr.iml
@@ -9,7 +9,7 @@
       <excludeFolder url="file://$MODULE_DIR$/results" />
       <excludeFolder url="file://$MODULE_DIR$/tb_logger" />
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.7 (pytorch-1.7)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="torch-cu11" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
     <orderEntry type="module" module-name="SwitchedConvolutions" />
   </component>
diff --git a/codes/train.py b/codes/train.py
index 8e60cea7..6c390eb9 100644
--- a/codes/train.py
+++ b/codes/train.py
@@ -32,7 +32,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_exd_imgset_ssgr.yml')
+    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_ssgr_constrained_gan.yml')
     parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
     parser.add_argument('--local_rank', type=int, default=0)
     args = parser.parse_args()