from model import Net
from perforatedai import utils_perforatedai as UPA

baseline = Net()
dendritic = UPA.perforate_model(Net())

def count(m):
    return sum(p.numel() for p in m.parameters())

print("Baseline parameters:", count(baseline))
print("Dendritic parameters:", count(dendritic))