torch version issue

[nrkarthikeyan]

Running the HELOC notebook produces this error:
module 'torch.jit' has no attribute '_script_if_tracing'
which is resolved by using a different version of pytorch
!pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

setup.py may need an update to use the correct version of torch and torchvision