search

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

Chunkwise performance #7

Closed BrianPulfer closed 1 year ago

BrianPulfer commented 1 year ago

Solves #5. Improvements are likely still possible (i.e. removal of for loop to get states).