search

FrancescoSaverioZuppichini / glasses

High-quality Neural Networks for Computer Vision 😎
https://francescosaveriozuppichini.github.io/glasses-webapp/
MIT License
437 stars 37 forks source link

ViT inplace backward error #207

Closed FrancescoSaverioZuppichini closed 3 years ago

FrancescoSaverioZuppichini commented 3 years ago

We need to replace https://github.com/FrancescoSaverioZuppichini/glasses/blob/ba5efd961cd3ebe87e754a3190f1ca931c8b9947/glasses/models/classification/vit/__init__.py#L149

with an inplace operation