models icon indicating copy to clipboard operation
models copied to clipboard

fix load pretrain and load params due to the shape check in fluid.io

Open SunGaofeng opened this issue 6 years ago • 1 comments

Fix load pretrain and load params due to the shape check in fluid.io added in Version 1.6.0 In StNet and Nonlocal, some of the pretrained params' shape is not the same as those in the program, this will lead to error in Version 1.6.0. This pr modified the loading process by creating vars in another scope and then load pertained params. When loading finished, the value will be transformed as set to the parameters of the target program.

SunGaofeng avatar Oct 13 '19 10:10 SunGaofeng

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

CLAassistant avatar Sep 30 '24 03:09 CLAassistant