[train_ds](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset") = [TensorDataset](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset")([x_train](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"), [y_train](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"))
train_dl = [DataLoader](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader "torch.utils.data.DataLoader")([train_ds](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset"), batch_size=bs, shuffle=True)

[valid_ds](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset") = [TensorDataset](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset")([x_valid](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"), [y_valid](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"))
valid_dl = [DataLoader](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader "torch.utils.data.DataLoader")([valid_ds](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset"), batch_size=bs * 2)