AIX360
AIX360 copied to clipboard
torch version issue
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