save

2023-03-03 17:06:11 +00:00
parent 3b414e76d3
commit 165adf7953
12 changed files with 400 additions and 20 deletions
@@ -5,12 +5,12 @@ import numpy as np

 if __name__ == '__main__':
    garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item()
-    folder = '/home/workspace/data/results/errors'
+    folder = '/home/workspace/data/results/errors/iccv'

-    checkpoint_name = 'abl_2_pyr_32_64_128_256_512'
-    checkpoint_name_skirt = 'abl_2_pyr_32_64_128_256_512_skirt'
-    iteration = '50000'
-    iteration_skirt : str = '50000'
+    checkpoint_name = 'iccv_pyr_32_64'
+    checkpoint_name_skirt = 'iccv_pyr_32_64_skirt'
+    iteration = '200000'
+    iteration_skirt : str = '235000'

    # checkpoint_name = 'abl_skin'
    # checkpoint_name_skirt = checkpoint_name + '_skirt'