def _getNNAttr(self, name: str):
    import torch.nn as NN
    ans = getattr(NN, name)
    return ans
def _getNNAttr(self, name: str):
    import torch.nn as NN
    ans = getattr(NN, name)
    return ans