Bug when batch size is 1 with Recurrent mode

BrianPulfer / vision-retention-networks

Unofficial reimplementation of ViR: Vision Retention Networks by Hatamizadeh et. al. (https://arxiv.org/abs/2310.19731)

Apache License 2.0

18 stars 1 forks source link

Open BrianPulfer opened 1 year ago

BrianPulfer commented 1 year ago

Currently, if the batch size is one, an error is generated when using the recurrent mode.