def method(model): return SmoothGrad(model, augment_by_n=2)
def method(model): return SmoothGrad(model)