From 87b7786b1e5a996c1929996df3296fe752985f49 Mon Sep 17 00:00:00 2001 From: Eric Bezzam Date: Wed, 21 Feb 2024 18:28:57 +0100 Subject: [PATCH] Add todo. --- scripts/classify/train_celeba_vit.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/classify/train_celeba_vit.py b/scripts/classify/train_celeba_vit.py index 2544f263..3d55bb0c 100644 --- 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