pytorch为了加快数据提取,定义建立train_data,train_loader
train_data = TensorDataset(torch.tensor(train_data, dtype=torch.int))
train_loader = DataLoader(train_data, batch_size= batch_size,shuffle=True, drop_last=True)
pytorch为了加快数据提取,定义建立train_data,train_loader
train_data = TensorDataset(torch.tensor(train_data, dtype=torch.int))
train_loader = DataLoader(train_data, batch_size= batch_size,shuffle=True, drop_last=True)
本文标题:pytorch各种操作
本文链接:https://www.haomeiwen.com/subject/jiuoyhtx.html
网友评论