venv
trained_pytorch.pth