print(sum(p.numel() for p in model.parameters()))    # 195
print(model)