pytorch

nn.Module

• self.register_buffer()，将该成员变量在前向传播前，保存下来（即保存在model.state_dict中）

torch.utils

data.Dataset

自定义数据集继承父类，自定义数据集需要重写__getitem__、__len__方法

data.DataLoader

• collate_fn：将自定义数据集__getitem__输出合成一个list，然后分离，各自成为一个batch_size的Tensor，对字典的处理比较特殊复杂；自定义collate_fn（unet代码中）
• pin_memory：锁页内存，开启后速度更快

tensorboard.Summarywriter

• 实例化writer对象

writer = SummaryWriter("./logs/dataloader")

• 添加log

writer.add_images("Epoch:{}".format(epoch), imgs, step)

• 启动tensorboard

tensorboard --logdir="./logs/dataloader" --bind_all

torchvision

datasets

自带一些数据集下载功能，不过此种方式定义的需要传入transform

tranforms

预处理，使用compose