Pytorch——DataLoader的学习笔记~

1 前言

今天在学习动物多分类~

示例代码里面使用的是Pytorch~

2 DataLoader数据载入器

2.1 num_workers参数设置

num_workers是DataLoader构造函数的参数，其描述为：how many subprocesses to use for data loading. ``0`` means that the data will be loaded in the main process.

看描述来说，跟子线程的数量有关系，那么可能跟CPU的线程数量有关，但是看到一些文献说，如果设置成非0的数量，则会导致GPU运算出现异常，那么到底该如何设置比较好呢？