Is it Possible to have Conditional Statements in a Network?

I am trying out a couple of variants of networks. Depending on the variant, I want to use certain specific layers and losses - different variants make use of different set of layers and losses.

Would it be safe to use ‘if…else’ conditional statements to select layers and losses depending on the variant I am trying out; so that I don’t have to create separate files/projects for different variants? (Network structure remains constant during the experimentation with a variant)

Yes, it would work because of the beauty of dynamic computation graph :slight_smile:

@SimonW Thanks a lot!