У меня есть какая-то модель в pytorch, с обновляемыми весами которой я хочу получить доступ и изменить вручную.
Как бы это было сделано правильно?
В идеале, я хотел бы получить тензор этих весов.
Мне кажется, что
for parameter in model.parameters():
do_something_to_parameter(parameter)
было бы неправильным путем, потому что
- Он не использует графический процессор и являетсяне в состоянии
- Он даже не использует низкоуровневую реализацию
Какой правильный способ доступа к весам модели вручную (не через loss.backward и optimizer.step)?