Skip to content

Commit 8f2f06b

Browse files
authored
Update 2024-5-10-use-of-dataloader-and-dataset.md
Fix Content Error
1 parent fbff551 commit 8f2f06b

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

_posts/2024-5-10-use-of-dataloader-and-dataset.md

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,6 @@ from torch.utils.data import DataLoader
5555
在主程序中,调用已经实现的 `Dataset` 子类来将其实例化。
5656

5757
```python
58-
from dataloader import MyDataset as Dataset
59-
6058
train_dataset = Dataset(source_arr, target_arr)
6159
```
6260
{: file="main.py" }
@@ -70,7 +68,7 @@ train_dataloader = DataLoader(train_dataset, arg1, arg2, ...)
7068
```
7169
{: file="main.py" }
7270

73-
与其在创建对象的时候填写参数,更推荐的是提前将参数打包成字典,在创建对象时进行解包:
71+
与其在创建 `DataLoader` 对象的时候填写参数,更推荐的是提前将参数打包成字典,在创建对象时进行解包:
7472

7573
```python
7674
dataloader_args = {"batch_size": 256, "shuffle": True, "num_workers": 8}

0 commit comments

Comments
 (0)