remove default hf_repo from config

This commit is contained in:
limiteinductive 2023-12-15 18:49:58 +01:00 committed by Benjamin Trom
parent 6a1fac876b
commit 12eef9cca5
2 changed files with 2 additions and 2 deletions

View file

@ -44,7 +44,7 @@ dropout_probability = 0
use_gyro_dropout = false
[dataset]
hf_repo = "acme/cat-toy"
hf_repo = "acme/images"
revision = "main"
horizontal_flip = true
random_crop = true

View file

@ -26,7 +26,7 @@ def load_hf_dataset(
class HuggingfaceDatasetConfig(BaseModel):
hf_repo: str = "finegrain/unsplash-dummy"
hf_repo: str
revision: str = "main"
split: str = "train"
horizontal_flip: bool = False