mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
17 Commits
136 Branches
65 Tags
47 MiB
Python 100%
 
Go to file
Ross Wightman d87824bd65 Merge branch 'master' of github.com:rwightman/pytorch-models 2019-03-17 09:57:36 -07:00
data Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models 2019-03-11 22:17:42 -07:00
models Merge branch 'master' of github.com:rwightman/pytorch-models 2019-03-17 09:57:36 -07:00
optim Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms 2019-03-10 14:23:16 -07:00
scheduler Resnext added, changes to bring it and seresnet in line with rest of models 2019-03-01 15:44:04 -08:00
inference.py Fixup validate/inference script args, fix senet init for better test accuracy 2019-02-22 14:07:50 -08:00
train.py Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models 2019-03-11 22:17:42 -07:00
utils.py Minor updates 2019-02-08 20:56:24 -08:00
validate.py Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models 2019-03-11 22:17:42 -07:00