Wandb and accelerate

2023-11-08 02:28:53 -08:00
parent a5f6dea673
commit 2d1a751441
3 changed files with 58 additions and 23 deletions
@@ -63,7 +63,7 @@ class SwappingDataset(data.Dataset):

    def preprocess(self):
        """Preprocess the Swapping dataset."""
-        print("processing Swapping dataset images...")
+        #print("processing Swapping dataset images...")

        temp_path   = os.path.join(self.image_dir,'*/')
        pathes      = glob.glob(temp_path)