diff --git a/elements/logger.py b/elements/logger.py
index 245f5ba..5036d96 100644
--- a/elements/logger.py
+++ b/elements/logger.py
@@ -319,6 +319,8 @@ def __call__(self, summaries):
         bystep[step][name] = wandb.Image(value)
       elif len(value.shape) == 4:
         assert value.shape[3] in [1, 3, 4], value.shape
+        if value.shape[3] == 1:  # Address "ValueError: Can't write images with one color channel." issue
+          value = np.concatenate((value,) * 3, axis=3)
         value = np.transpose(value, [0, 3, 1, 2])
         if value.dtype != np.uint8:
           value = (255 * np.clip(value, 0, 1)).astype(np.uint8)