File size: 303 Bytes
347b44e |
1 2 3 4 5 6 7 8 9 10 11 12 |
def count_parameters(model):
return sum(p.numel() for p in model.parameters() if p.requires_grad)
def count_conv_layers(model):
cnt = 0
for mo in model.modules():
if type(mo).__name__ == 'Conv2d':
cnt += 1
print(type(model).__name__, cnt, count_parameters(model))
|