Change where images weres stored

.gitignore

@@ -1,3 +1,4 @@
 .DS_Store
 .vscode/
 pyrightconfig.json
+*.jpg

src/config.py

@@ -4,7 +4,7 @@ import pydantic
 
 MAX_DOWNLOAD_TIME = 0.2
 
-IMAGE_DOWNLOAD_PATH = pathlib.Path("/tmp/images")
+IMAGE_DOWNLOAD_PATH = pathlib.Path("./data/images")
 
 
 class DataConfig(pydantic.BaseModel):
@@ -30,7 +30,7 @@ class ModelConfig(pydantic.BaseModel):
 
 class TrainerConfig(pydantic.BaseModel):
     epochs: int = 20
-    batch_size: int = 256
+    batch_size: int = 64
     learning_rate: float = 5e-4
     accumulate_grad_batches: int = 1
     temperature: float = 1.0

src/data.py

@@ -102,7 +102,8 @@ if __name__ == "__main__":
     )
     train_dl, valid_dl = get_dataset(transform, tokenizer, hyper_parameters)
 
+    batch = next(iter(train_dl))
+    print({k: v.shape for k, v in batch.items()})  # torch.Size([1, 3, 128, 128])
+
     for batch in tqdm(train_dl):
         continue
-
-    print("hellow")