Как получить доступ к параметрам модели Pytorch по индексу

Question

Если у меня есть сеть, скажем, с 10 слоями, включая смещения, как я могу получить доступ к параметрам i-го слоя только по индексу?

В настоящее время я делаю что-то вроде этого

for parameter in myModel.parameters():
    parameter.data /= 5 

Как я могу получить доступ к параметру .data с индексом?Например, я хотел бы получить доступ к 9-му слою без итераций, например, myModel.parameter.data [8] или что-то подобное.

Answer 1

просто выполните:

layers=[x.data for x in myModel.parameters()]

Теперь это будет список весов и смещений, чтобы получить доступ к весам первого слоя, которые вы можете сделать:

print(layers[0])

вЧтобы получить доступ к смещениям первого уровня:

print(layers[1])

и т. д.

Помните, что для любого конкретного слоя смещение имеет значение false, например, если смещениеложно для второго слоя, тогда layers[3] фактически даст веса третьего слоя.