CUDA out of memory in WSL2

[Enchante503]

Change "pin_memory=True" to "pin_memory=False" in train.py

train_loader = DataLoader(
        train_dataset,
        num_workers=4,
        shuffle=False,
        pin_memory=False,
        collate_fn=collate_fn,
        batch_sampler=train_sampler,
        persistent_workers=True,
        prefetch_factor=8,
    )