Koliko težina se trenira u sloju konvolucije sa jezgrom 5x5 i 10 filtera?

U sloju konvolucije sa jezgrom 5x5 i 10 filtara, broj treniranih težina računa se kao: veličina jezgra pomnožena sa brojem filtara.

Ako pretpostavimo da je ulazni kanal jedan (npr. crno-bela slika), onda:

5 × 5 × 10 = 250 težina.

Ako postoji više ulaznih kanala (npr. 3 za RGB), formula će biti:

5 × 5 × broj_ulaznih_kanala × 10.

Pored toga, svaki filtar obično ima pomak (bias), što znači još 10 parametara (po jedan za svaki filtar).

Primer za 3 ulazna kanala:

kernel_size = 5
input_channels = 3
filters = 10
weights = kernel_size * kernel_size * input_channels * filters  # 5*5*3*10 = 750
biases = filters  # 10
total_params = weights + biases  # 760