* [Fix] Fix vit init bug * Add some vit unit tests * Modify module import * Fix pretrain weights bug * Modify pretrained judge * Add some unit tests to improve code cov * Optimize code * Fix vit unit test |
||
---|---|---|
.. | ||
__init__.py | ||
drop.py | ||
inverted_residual.py | ||
make_divisible.py | ||
res_layer.py | ||
se_layer.py | ||
self_attention_block.py | ||
timm_convert.py | ||
up_conv_block.py |