avans06's picture
1. Add weights_only=True parameter to torch.load to avoid FutureWarning.
7737b56