PyTorch v2.4.0반영, pytorch/tutorials@030a8e8(PyTorchKorea/pytorch.kr#54)

Co-authored-by: Taegeun Kwack <[email protected]>
rhkr8521 · Sep 9, 2024 · a0aa2c9 · a0aa2c9
1 parent 3bb3ada
commit a0aa2c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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