Improve ToRGB docs

Author: ProGamerGov

captum/optim/_param/image/transforms.py

@@ -77,9 +77,11 @@ class ToRGB(nn.Module):
     We offer two precalculated transforms: Karhunen-Loève (KLT) and I1I2I3.
     KLT corresponds to the empirically measured channel correlations on imagenet.
     I1I2I3 corresponds to an approximation for natural images from Ohta et al.[0]
-    If you wish to calculate your own KLT transform matrix on a custom dataset,
-    then please see :func:`captum.optim.dataset.dataset_klt_matrix` for an example
-    of how to do so.
+
+    While the default transform matrices should work for the vast majority of use
+    cases, you can also use your own 3x3 transform matrix. If you wish to calculate
+    your own KLT transform matrix on a custom dataset, then please see
+    :func:`captum.optim.dataset.dataset_klt_matrix` for an example of how to do so.
 
     [0] Y. Ohta, T. Kanade, and T. Sakai, "Color information for region segmentation,"
     Computer Graphics and Image Processing, vol. 13, no. 3, pp. 222–241, 1980