Skip to content

Commit

Permalink
Add/type annotations/model vision (#16151)
Browse files Browse the repository at this point in the history 
* add types annotations for Beit (PyTorch)

* add types annotations for ViT (PyTorch)

* add types annotations for Deit (PyTorch)

* change Optional[bool] to bool into some places at Beit

* change Optional[bool] to bool into some places at ViT
  • Loading branch information
@johnnv1 @FrancescoSaverioZuppichini
johnnv1 authored and FrancescoSaverioZuppichini committed Mar 21, 2022
1 parent e06c415 commit f8514f9
Show file tree
Hide file tree
Showing 5 changed files with 299 additions and 214 deletions.
Loading

0 comments on commit f8514f9

Please sign in to comment.