Cleaner PyTorch training code that scales to more GPUs.
Recent revisions are kept. Restoring one creates a new revision.