search

lucidrains / bottleneck-transformer-pytorch

Implementation of Bottleneck Transformer in Pytorch
MIT License
671 stars 82 forks source link

About dim_out? #5

Closed Zengpeizhi closed 3 years ago

Zengpeizhi commented 3 years ago

Why dim_out (out_put channel must be 2048? when i set dim_out to ohters,it didn't work.

lucidrains commented 3 years ago

@Zengpeizhi fixed in the latest! https://github.com/lucidrains/bottleneck-transformer-pytorch/commit/1c04fb3c46a280037a395f8f7ac22bf610fa2a64