search

JeiKeiLim / kindle

Making a PyTorch model easier than ever!
https://limjk.ai/kindle/
MIT License
78 stars 7 forks source link

Yaml custom module width multiply #9

Closed JeiKeiLim closed 3 years ago

JeiKeiLim commented 3 years ago

Currently, custom module from yaml only supports width multiplier in global level. It might be good idea to take an argument of channel multiplier in module level.