本文以mnist介绍dateset,并介绍了如何自定义dataset。

mnist代码: 下载

可以看到,一个dataset都要继承一个torch.utils.data内的Dataset类。为了使一个dataset可以被pytorch内置的dataloader载入,必须实现两个method:

    def __getitem__(self, index):
    def __len__(self):

第一个__getitem__通过一个index来返回一个数据项,第二个__len__返回数据的数目。

笔者的环境大概是这样的,家里有一台NAS常年开机,一台装有显卡的台式机有时候需要启动来跑一些深度学习的任务,最近做了一个微信机器人号,基于itchat,可以随时远程打开电脑。代码如下:

Read on →