Fix NotImplementedError

✅ Solution

This error often arises when trying to directly move a `meta` device tensor (a placeholder without actual data) to another device like CUDA. Instead of `.to("cuda")`, use `torch.nn.Module.to_empty(device="cuda")` to properly initialize the module's weights on the target device, or handle data loading before moving tensors. Ensure all parts of your model and data are correctly initialized on the expected device before any computations.