Minor Changes

Author: pranavsinghps1

configs/config/benchmark/linear_image_classification/imagenet1k/byol_transfer_in1k_linear.yaml

@@ -1,4 +1,6 @@
 # @package _global_
+# BYOL:Bootstrap Your Own Latent - https://arxiv.org/abs/2006.07733
+# Configuration based on BYOL paper - Appendix C -- C.1
 config:
   VERBOSE: True
   LOG_FREQUENCY: 200

vissl/losses/byol_loss.py

@@ -41,7 +41,9 @@ class BYOLLoss(ClassyLoss):
     """
     This is the loss proposed in BYOL 
     - Bootstrap your own latent: (https://arxiv.org/abs/2006.07733)
-    This function computes the loss and restores loss from checkpoints.
+    This class wraps functions which computes
+    - loss
+    - restores loss from checkpoints.
 
     Config params:
         embedding_dim (int): head output output dimension