Deep Learning Note
PyTorch
What is PyTorch?
- An machine learning framework in Python.
- Two main features: ○ N-dimensional Tensor computation (like NumPy) on GPUs ○ Automatic differentiation for training deep neural networks
Dataset & Dataloader
Dataset
stores data samples and expected values
dataset = MyDataset(file)Dataloader
groups data in batches, enables multiprocessing
dataloader = DataLoader(dataset, batch_size, shuffle=True)
- 本文作者： NICK
- 本文链接： https://nicccce.github.io/CourseNotes/Deep-Learning-Note/
- 版权声明： 本博客所有文章除特别声明外，均采用 MIT 许可协议。转载请注明出处！