常用方法strip()
train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=batch_size, # num_workers=1, shuffle=True, pin_memory=True, collate_fn=train_dataset.collate_fn)
pin_memory=True:数据固定存储在内存中,不会被虚拟内存机制换出到磁盘
pin_memory=False:数据存储在虚拟内存中,可能会被虚拟内存机制换出到磁盘