test-16

inference_custom.py

@@ -9,7 +9,7 @@ from dataset import DatasetSimple
 from load_models import load_generator
 from tf_utils import allow_memory_growth
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "3"
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
 def m2uv_array(m2uv):
     result = np.zeros((14475, 2), dtype=np.int32)
@@ -83,9 +83,9 @@ def main():
     checkpoints = [
         {
             'res': 512,
-            'ckpt_dir': 'models/stylegan2-test-12-512x512',
+            'ckpt_dir': 'models/stylegan2-test-15-512x512',
             'use_custom_cuda': True,
-            'folder': 'output/test-12-140000',
+            'folder': 'output/test-15-115000',
             'data_folder': '/home/workspace/data/dataset/stylegan_new/train'
         },
     ]