move FCN output layer initialization out of the for loop

JJGO / hyperlight

Modular and intuitive Hypernetworks in Pytorch

Apache License 2.0

32 stars 2 forks source link

Open BailiangJ opened 9 months ago

BailiangJ commented 9 months ago

Hi @JJGO ,

thanks a lot for sharing the code!

If I understand correctly, this line should be outside the for loop?