What is the difference between torch.save() and torch.utils.checkpoint.checkpoint()
Hi,
The first one can be used to save some data in a file on disk.
The second one is used to perform a training trick to reduce the memory usage.
You can check the doc for each for more details.