Cian-H/symbolic_nn_tests
1
0
Fork 0
mirror of https://github.com/Cian-H/symbolic_nn_tests.git synced 2025-12-22 14:11:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated dataloader to default to tensor transform

Browse Source
This commit is contained in:
Cian Hughes
2024-05-13 15:55:54 +01:00
parent 02d70964e1
commit 63bea4d355
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
symbolic_nn_tests/dataloader.py
View File

@@ -12,8 +12,11 @@ def get_dataset(
split: (float, float, float) = (0.7, 0.1, 0.2),
batch_size: int = 128,
drop_last: bool = False,
**kwargs,
):
ds = Caltech256(PROJECT_ROOT / "datasets/", download=True, transform=ToTensor())
_kwargs = {"transform": ToTensor()}
_kwargs.update(kwargs)
ds = Caltech256(PROJECT_ROOT / "datasets/", download=True, **_kwargs)
train, test, val = (
BatchSampler(i, batch_size, drop_last) for i in random_split(ds, split)
)