Updated PyTorch to version 2.3.1+cu121 and TorchVision to 0.18.1+cu121 due to security reasons.
b365d37
Aryeh Rotberg
commited on