Torch学习记录

pytorch有一些基础函数须要了解,在本文章记录一下。web torch.utils.data.DataLoader DataLoader是导入图片的操做,里面有一些参数,好比batch_size和shuffle等,默认load进去的图片类型是PIL.Image.open的类型网络 torchvision.transforms torchvision.transforms里面的操做是对导入的图片
相关文章
相关标签/搜索