LightGlue Params

[DiZhang980]

How to determine the parameter quantity of each layer for LightGlue? @fabio-sim

[fabio-sim]

Hi @DiZhang980, thank you for your interest in LightGlue-ONNX.

Assuming you mean each transformer layer, you can add this print statement:

print(f"Layer {i+1}: {sum(p.numel() for p in self.transformers[i].parameters())} parameters")

to the following loop:

LightGlue-ONNX/lightglue_onnx/lightglue.py

Lines 328 to 330 in e14a259

	for i in range(self.conf.n_layers):
	# self+cross attention
	desc0, desc1 = self.transformers[i](desc0, desc1, encoding0, encoding1)

I get the following output:

Layer 1: 1250560 parameters
Layer 2: 1250560 parameters
Layer 3: 1250560 parameters
Layer 4: 1250560 parameters
Layer 5: 1250560 parameters
Layer 6: 1250560 parameters
Layer 7: 1250560 parameters
Layer 8: 1250560 parameters
Layer 9: 1250560 parameters

[DiZhang980]

Thank you very much for your reply. I will try your statement.