Tensorflow 2.x: как вручную назначить веса свертки с помощью numpy

В tensorflow 1.x это можно сделать с помощью граф и сессия, что довольно утомительно.

Есть ли более простой способ вручную назначить предварительно обученные веса для конкретной свертки в tensorflow 2.x?


tensorflow tensorflow2.0 onnx
person mcExchange    schedule 15.06.2020    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Если вы работаете с Keras внутри Tensorflow 2.x, у каждого слоя есть метод с именем set_weights, который вы можете использовать для замены весов или назначения новых из массивов Numpy.

Скажем, например, что вы занимаетесь перегонкой знаний. Затем вы можете назначить веса учителя ученику:

conv.set_weights(teacher.convx.get_weights())

где conv — особый слой ученика, а convx — гомолог учителя.

Вы можете проверить документацию для более подробной информации:

Документация — set_weights()

person BCJuan    schedule 06.06.2021