Checkpoint at epoch 0 will be covered in the end when using DDP

[yankungou]

Hi Spijkervet, thank you for your code. I found that the checkpoint at epoch 0 will be covered in the end when using DDP, maybe it's better to add a line before this line.

    if args.nr == 0:
        save_model(args, model, optimizer)