La Data Science

Tout sur la data

Compter le Nombre de Paramètres d’un Modèle PyTorch

Problème :

Comment savoir quel est le nombre de paramètres d’un modèle PyTorch?

Solution :

Utiliser la ligne de code suivante:

	total_parameters = pytorch_total_params = sum(p.numel() for p in model.parameters())
	print(f"Nombre de paramètres : {total_parameters:,}")

	# model est un modèle que vous avez déjà défini à l'avance.

Compléments :

Jean MP