From a0aa2c98a061e9f55c4df84bfdffe498eccdca5a Mon Sep 17 00:00:00 2001 From: Juhyeon Lee <81895590+gogori6565@users.noreply.github.com> Date: Mon, 9 Sep 2024 17:47:40 +0900 Subject: [PATCH] =?UTF-8?q?PyTorch=20v2.4.0=EB=B0=98=EC=98=81,=20pytorch/t?= =?UTF-8?q?utorials@030a8e8(PyTorchKorea/pytorch.kr#54)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Taegeun Kwack --- beginner_source/basics/data_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py index 2baef464..c82cd2b0 100755 --- a/beginner_source/basics/data_tutorial.py +++ b/beginner_source/basics/data_tutorial.py @@ -125,7 +125,7 @@ class CustomImageDataset(Dataset): def __init__(self, annotations_file, img_dir, transform=None, target_transform=None): - self.img_labels = pd.read_csv(annotations_file, names=['file_name', 'label']) + self.img_labels = pd.read_csv(annotations_file) self.img_dir = img_dir self.transform = transform self.target_transform = target_transform