You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Ross Wightman
58571e992e
Change block avgpool in senets to mean for performance issues with NVIDIA and AMP especially
|
6 years ago |
| .. |
|
__init__.py
|
Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms
|
6 years ago |
|
adaptive_avgmax_pool.py
|
Add some models, remove a model, tweak some models
|
6 years ago |
|
densenet.py
|
Add some models, remove a model, tweak some models
|
6 years ago |
|
dpn.py
|
Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models
|
6 years ago |
|
fbresnet200.py
|
Initial commit, puting some ol pieces together
|
6 years ago |
|
inception_resnet_v2.py
|
Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models
|
6 years ago |
|
inception_v4.py
|
Initial commit, puting some ol pieces together
|
6 years ago |
|
median_pool.py
|
Initial commit, puting some ol pieces together
|
6 years ago |
|
model_factory.py
|
Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models
|
6 years ago |
|
pnasnet.py
|
Initial commit, puting some ol pieces together
|
6 years ago |
|
resnet.py
|
Add some models, remove a model, tweak some models
|
6 years ago |
|
resnext.py
|
Merge branch 'master' of github.com:rwightman/pytorch-models
|
6 years ago |
|
senet.py
|
Change block avgpool in senets to mean for performance issues with NVIDIA and AMP especially
|
6 years ago |
|
xception.py
|
Initial commit, puting some ol pieces together
|
6 years ago |
