pytorch dataset dataloader

pytorch提供了方便的接口,在实操环境中,你只需要:

  1. 创建一个自定义的Dataset对象
  2. 赋值给内置的DataLoader,用于为训练模型提供batch。

你可能感兴趣的:(数据结构与算法,pytorch)