search

er-muyue / DeFRCN

MIT License
181 stars 43 forks source link

The weight of Affine Matrix #32

Closed MartinYYYYan closed 2 years ago

MartinYYYYan commented 2 years ago

Thanks for your great contribution!

I am very curious about the Affine Matrix. I found that the weights of Affine Matrix are not saved in the trained model. Whenever I re-load the trained model, the value of the affine matrix is 1 (initial value).

Am I doing something wrong? Or the Affine Matrix is only used for weighted back-propagation?

MartinYYYYan commented 2 years ago

Sorry, I have loaded the wrong pre-trained model.