Update PFE layer modules to handle operators
Closed this issue · 1 comments
timmoon10 commented
The FC and conv modules have an option to pass in a layer type, which will be applied as an activation function whenever the module is called. However, PR #1955 has removed many basic layers (notably tanh) and replaced them with operators. I think we should update these modules to accommodate any callable object as an activation function.