Update ema.py

[sagarishere]

Remove the unnecessary if-else block for checking whether the module is a nn.DataParallel module. Since the shadow parameters are stored in a dictionary using the name of the parameters, this check is not necessary.