Как посмотреть параметры модели пироха?

Question

У меня есть простая нейронная сеть Pytorch, которую я скопировал из openai , и я до некоторой степени изменил ее (в основном входные данные).

Когда я запускаю свой код, выходные данные сети остаются одинаковыми в каждом эпизоде, как будто не происходит обучения.

Я хочу посмотреть, происходит ли какое-либо обучение, или же по какой-то другой причине результаты будут такими же.

Как я могу убедиться, что любое движение происходит с весами?

Спасибо

Answer 1

Зависит от того, что вы делаете, но проще всего было бы проверить вес вашей модели.

Вы можете сделать это (и сравнить с предыдущими итерациями), используя следующий код:

for parameter in model.parameters():
    print(parameter.data)

Если веса меняются, нейронная сеть оптимизируется (что не обязательно означает, что она узнает что-то полезное в частности).