search

assafshocher / ZSSR

"Zero-Shot" Super-Resolution using Deep Internal Learning
Other
401 stars 102 forks source link

wieght initiliazing #35

Open elliotTAU opened 3 years ago

elliotTAU commented 3 years ago

Hi

Is it possible to initialize weights with your build function "build_network" or "init_sess" ? Or i need to write full function to do so?

I want to perform fine-tunning/transfer learning So maybe just initializing the last layer

thanks

assafshocher commented 3 years ago

Hi, The weights are initialized at https://github.com/assafshocher/ZSSR/blob/d2948526fde799a0c111469b78cd9e1b8659cb0f/ZSSR.py#L146

So you would need to add some flag and an option to load existing weights