Yes, I see that you are using Python, but this discussion forum is specialized on PyTorch (the ML framework). While we can certainly try to help solving plain Python issues, the majority of the discussion is focused on PyTorch, which is why I suggested other boards for a faster response