class_docstr: Adds PyText support for
  Decoupled Weight Decay Regularization for Adam as done in the paper:
  https://arxiv.org/abs/1711.05101
  for more information read the fast.ai blog on this optimization
  method here: https://www.fast.ai/2018/07/02/adam-weight-decay/
class_docstr_tok: ["Adds","PyText","support","for","Decoupled","Weight","Decay","Regularization","for","Adam","as","done","in","the","paper",":","https",":","\/\/arxiv.org\/abs\/1711.05101","for","more","information","read","the","fast.ai","blog","on","this","optimization","method","here",":","https",":","\/\/www.fast.ai\/2018\/07\/02\/adam-weight-decay\/"]
code: def step(self, *args, **kwargs):
      # this is to enable FP16 training, following https://fb.workplace.com/groups/300451907202972/permalink/804122046835953/
      if "groups" in kwargs.keys():
          kwargs.pop("groups")
      return super().step(*args, **kwargs)
  
imports: import torch
  from fairseq.utils import clip_grad_norm_
  from pytext.config import ConfigBase
  from pytext.config.component import Component, ComponentType
label_desc: Use image node_id 3 for calling the AdamW obj's underlying member method code with example usage: obj.step() and returns: super