[Solved] Pytorch DataLoader: UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow – Pytorch

The Problem:

When using PyTorch’s DataLoader() to train a model, a User Warning is displayed indicating that creating a tensor from a list of numpy arrays is slow. The warning suggests converting the list to a single numpy array before creating the tensor. The user wants to explore options to convert the DataLoader() dataset to a tensor or find a better approach to resolve this issue.

The Solutions:

Solution 1: Convert to Tensor using `torch.from_numpy()`

To resolve the warning, modify the code segment that creates the tensor from a list of numpy arrays:

labels = torch.from_numpy(labels)