From 7da57ec434f2b817f36d0e5bd6b6449537f3c311 Mon Sep 17 00:00:00 2001
From: Yoshitomo Matsubara <yoshitom@uci.edu>
Date: Sat, 11 Nov 2023 14:23:55 -0800
Subject: [PATCH] Fix a typo

---
 torchdistill/losses/registry.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchdistill/losses/registry.py b/torchdistill/losses/registry.py
index b6722f2b..05d304ef 100644
--- a/torchdistill/losses/registry.py
+++ b/torchdistill/losses/registry.py
@@ -180,7 +180,7 @@ def _register_func2extract_model_output(func):
 
 def get_loss(key, **kwargs):
     """
-    Gets a registered dataset module.
+    Gets a registered (low-level) loss module.
 
     :param key: unique key to identify the registered loss class/function.
     :type key: str