How to load the pth file for DistributedDataparallel model?

Question

I trained the model with DistributedDataParallel and made the pth file

  if args.gpu is not None:
            print(\'Gpu setting...\',args.gpu)