Could you install the latest PyTorch and torchvision releases and check, if you are still facing the same issue?
PyTorch 1.4.0 is quite old by now and I assume you are also using the corresponding torchvision version.