Merge pull request #82 from chairc/dev

Modify check_and_create_dir function; Modify comment.

tools/train.py

@@ -213,7 +213,7 @@ def train(rank=None, args=None):
         for i, (images, labels) in enumerate(pbar):
             # The images are all resized in dataloader
             images = images.to(device)
-            # Generates a tensor of size images.shape[0] * images.shape[0] randomly sampled time steps
+            # Generates a tensor of size images.shape[0] randomly sampled time steps
             time = diffusion.sample_time_steps(images.shape[0]).to(device)
             # Add noise, return as x value at time t and standard normal distribution
             x_time, noise = diffusion.noise_images(x=images, time=time)

utils/utils.py

@@ -146,6 +146,5 @@ def check_and_create_dir(path):
     :param path: Create path
     :return: None
     """
-    logger.warning(msg=f"Folder '{path}' is not exist.")
+    logger.info(msg=f"Check and create folder '{path}'.")
     os.makedirs(name=path, exist_ok=True)
-    logger.info(msg=f"Successfully create folder '{path}'.")

webui/web.py

@@ -180,7 +180,7 @@ def train(self, seed, conditional, sample, network, run_name, epochs, batch_size
                     return
                 # The images are all resized in dataloader
                 images = images.to(device)
-                # Generates a tensor of size images.shape[0] * images.shape[0] randomly sampled time steps
+                # Generates a tensor of size images.shape[0] randomly sampled time steps
                 time = diffusion.sample_time_steps(images.shape[0]).to(device)
                 # Add noise, return as x value at time t and standard normal distribution
                 x_time, noise = diffusion.noise_images(x=images, time=time)