save all

scripts/upscaler_train.py

@@ -11,7 +11,7 @@ parser.add_argument('--debug', action='store_true')
 args = vars(parser.parse_args())
 
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
-os.environ["CUDA_VISIBLE_DEVICES"] = '2'
+os.environ["CUDA_VISIBLE_DEVICES"] = '3'
 
 
 import time
@@ -39,10 +39,10 @@ from utils.utils import InfiniteDataLoader as DataLoader
 from pytorch_msssim import MS_SSIM
 
 
-low_res = 64
-is_skirt = False
-is_cape = True
-is_male = True
+low_res = 256
+is_skirt = True
+is_cape = False
+is_male = False
 skirt_modifier = '_skirt' if is_skirt else ''
 cape_modifier = '_cape' if is_cape else ''
 gender_modifier = '' if not is_cape else ('_m' if is_male else '_f')
@@ -244,6 +244,9 @@ def train():
 
         step += 1
 
+        if step > 100000:
+            break
+
 
 if __name__ == '__main__':
     if not args['debug']: