File size: 301 Bytes
2a2ae9a |
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(model.__name__, cnt, count_parameters(model()))
|