Weight replace manually

Hi, I wanted to add fully connected layer and weight initialization with weight manually.

Below are what I wanted to do.

  1. Save fully connected layer’s weights.
  2. Trained model A.
  3. In inference phase, add fully connected layer and replace fully connected layer’s weights to saved weights.

Is there any similar topic or reference code?