ASAM
ASAM copied to clipboard
Multi-GPU training
Hi, thanks for your open-source implementation of ASAM. I am wondering whether if ASAM support multi-GPU training, and how to calculate grad norm when using multiple GPUs? Thanks again and looking forward to your reply.