total_num = sum(p.numel() for p in model.parameters()) trainable_num = sum(p.numel() for p in model.parameters() if p.requires_grad) print('Total ',total_num) print('Trainable ',trainable_num)
