Référence de l'espace de nommage aidge_in_space.profiling

Coût de la partie embarquée, couche par couche : calcul, paramètres, mémoire d'activation. Plus de détails...

Fonctions
	profile_module (module, input_shape)
	Profil couche par couche d'un module, obtenu par une passe sur une entrée nulle.
	summarize (df)
	Totaux d'un profil de profile_module().

Description détaillée

Coût de la partie embarquée, couche par couche : calcul, paramètres, mémoire d'activation.

Documentation des fonctions

profile_module()

aidge_in_space.profiling.profile_module	(		module,
			input_shape )

Profil couche par couche d'un module, obtenu par une passe sur une entrée nulle.

Une ligne par convolution, convolution transposée et (I)GDN ; la conv 1x1 interne d'une FrozenGDN est comptée avec sa GDN.

Paramètres

module	Module à profiler (les poids n'ont pas d'importance).
input_shape	Forme de l'entrée, (1, C, H, W).

Renvoie

DataFrame avec les colonnes couche, type, entrée, sortie, paramètres, MAC (M), sqrt+div (M), entrée fp32 (Mo), sortie fp32 (Mo) et % MAC.

summarize()

aidge_in_space.profiling.summarize

(

df

)

Totaux d'un profil de profile_module().

Paramètres

df	Profil couche par couche.

Renvoie

Dictionnaire : GMAC, sqrt+div (M), paramètres, poids fp32 (Mo), poids int8 (Mo), plus grande activation fp32 (Mo) et pic entrée+sortie fp32 (Mo). Ce dernier (entrée + sortie de la couche la plus gourmande) est une borne basse de la mémoire de travail.