Add todo.

LCAV · Feb 21, 2024 · 87b7786 · 87b7786
1 parent 734f153
commit 87b7786
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/classify/train_celeba_vit.py b/scripts/classify/train_celeba_vit.py
@@ -42,6 +42,8 @@
 Other hyperparameters for classification can be found in
 `configs/train_celeba_classifier.yaml`.
 
+# TODO: update with Hugging Face dataset: https://huggingface.co/datasets/bezzam/DigiCam-CelebA-10K
+
 """
 
 import warnings