search

fastai / fastai_dev

fast.ai early development experiments
Apache License 2.0
641 stars 351 forks source link

Cat Layer #237

Closed tcapelle closed 5 years ago

tcapelle commented 5 years ago

This PR add a Concatenate layer equivalent to keras.layers.concatenate to concat the outputs of a list of layers. This type of layers is useful for architectures of type inception where different kernel sized convs are stacked together. I inherited from nn.ModueList but it could also inherit from fastai Module.

review-notebook-app[bot] commented 5 years ago

Check out this pull request on  ReviewNB

You'll be able to see Jupyter notebook diff and discuss changes. Powered by ReviewNB.

sgugger commented 5 years ago

Looks good to me. I like the inheritance from nn.ModuleList. Thanks!