Checkpoint: OverflowError: cannot serialize a string larger than 4GiB
🐛 Bug

Model checkpointing fails with the error: OverflowError: cannot serialize a string larger than 4GiB and halts training

PyTorch Version (e.g., 1.0): 1.5
OS (e.g., Linux): Linux
How you installed PyTorch (conda, pip, source): conda
Build command you used (if compiling from source):
Python version: 3.7
CUDA/cuDNN version: 10.1
GPU models and configuration: GTX 2080Ti
Any other relevant information:

Additional context
This is a known Python issue pytorch/pytorch#12085
Possible fix, set the protocol correctly