in quant/models/resnet.py [0:0]
def __init__(
self,
loss_fn: Callable[..., torch.Tensor],
block: str,
layer0: dict,
layer1: dict,
layer2: dict,
layer3: dict,
layer4: Optional[dict],
nonlins: List[str],
num_blocks: List[int],
output_classes: int,
moving_average_mode: str = 'off',
moving_average_momentum: float = 0.99,