520 Commits (e3f58fc90c144bc4e14ec29e8e98508a7f0f4106)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	e545bb9401	Update test_inference.py ... Not six min	5 years ago
Ross Wightman	305a2db705	Update test_inference.py ... Make the timeout 5-min for now, see if we can get a pass...	5 years ago
Ross Wightman	022ed001f3	Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options...	5 years ago
Ross Wightman	14e01b878c	Merge pull request #143 from michalwols/master ... Setup Github Action to instantiate and run a forward pass with each registered model.	5 years ago
michal	8c77f14cae	Install cpu version of torch on ubuntu	5 years ago
michal	69d725c9fe	Basic forward pass test for all registered models	5 years ago
michal	8da43e0617	Install extra dependencies required by some models and log test durations	5 years ago
Mike	16f1f77b41	Add a test workflow for github actions	5 years ago
Vyacheslav Shults	a7ebe09029	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given. ... Make small code refactoring	5 years ago
Ross Wightman	6cc11a8821	Merge pull request #141 from Animatory/fix_HRNet ... Fixed HRNet modules	5 years ago
Vyacheslav Shults	f0eb021620	Replace all None by nn.Identity() in HRNet modules	5 years ago
Ross Wightman	3b72ebff51	Merge pull request #140 from yoniaflalo/PR_MultiEpochsDataLoader ... added MultiEpochsDataLoader	5 years ago
AFLALO, Jonathan Isaac	a7f570c9b7	added MultiEpochsDataLoader	5 years ago
Ross Wightman	8d8677e03b	Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.	5 years ago
Ross Wightman	353a79aeba	Update README.md	5 years ago
Ross Wightman	c9b6f415ca	Bump version for Pypi release	5 years ago
Ross Wightman	31f4c126ec	Add EfficientNet pruned models to results files	5 years ago
Ross Wightman	375f3e5321	Fix pruned txt files not being installed during pip install	5 years ago
Ross Wightman	8ec554b82e	Merge pull request #136 from yoniaflalo/adding_effnet_pruned ... added efficientnet pruned weights	5 years ago
AFLALO, Jonathan Isaac	9c15d57505	added efficientnet pruned weights	5 years ago
Ross Wightman	a4d20a1cb8	Update README.md	5 years ago
Ross Wightman	ea300709f0	Fix model create fn not passing num_classes through. Fix #135	5 years ago
Ross Wightman	779cb0fcc0	Update results with new models	5 years ago
Ross Wightman	2c438c4e0d	Merge branch 'master' of github.com:rwightman/pytorch-models	5 years ago
Ross Wightman	28739bb721	Merge branch 'VRandme-blur'	5 years ago
Ross Wightman	2681a8d618	Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop	5 years ago
Ross Wightman	9590f301a9	Merge branch 'blur' of https://github.com/VRandme/pytorch-image-models into VRandme-blur	5 years ago
Ross Wightman	a5220ad3aa	Merge pull request #125 from Separius/patch-1 ... fix typo in eca	5 years ago
Ross Wightman	7a9942a75e	Merge branch 'yoniaflalo-adding_ECA_resnet'	5 years ago
Ross Wightman	0834fbc01c	Move pruned model adapt strings to separate txt files. A few minor formatting alignment tweaks.	5 years ago
Ross Wightman	ba793f5c1c	Merge branch 'adding_ECA_resnet' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_ECA_resnet	5 years ago
AFLALO, Jonathan Isaac	07f19dd699	added eca resnet	5 years ago
Ross Wightman	1d4ac1b165	Bump version for pypi release. Fix #130	5 years ago
Ross Wightman	be7c784d21	Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference	5 years ago
Ross Wightman	e3a98171b2	Remove some CRLF that made it into TResNet merge	5 years ago
Ross Wightman	0004f37d25	Some TResNet cleanup. ... * allow use of global pool arg, test-time-pooling * clean checkpoints to just contain state dict, add 448 res checkpoints * support DataParallel via lazy filter creation for JIT Downsample * some minor formatting (mostly alignment) preferences	5 years ago
Ross Wightman	64fe37d008	Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes #126	5 years ago
Ross Wightman	02a30411ad	Replace fp16 with amp support for validate.py script	5 years ago
Ross Wightman	e6f24e5578	Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults	5 years ago
Sepehr Sameni	20290b5b56	fix typo in eca	5 years ago
Ross Wightman	ebf82b84ac	Merge pull request #122 from mrT23/master ... TResNet models	5 years ago
Ross Wightman	e15f979457	Merge pull request #123 from aclex/mobilenetv3_fix_feature_extraction ... Merge changes in feature extraction interface to MobileNetV3	5 years ago
Alexey Chernov	bdb165a8a4	Merge changes in feature extraction interface to MobileNetV3 ... Experimental feature extraction interface seems to be changed a little bit with the most up to date version apparently found in EfficientNet class. Here these changes are added to MobileNetV3 class to make it support it and work again, too.	5 years ago
talrid	8a63c1add8	finalizing	5 years ago
talrid	6209146738	TResNet models	5 years ago
mrT23	98a7403ed4	Merge pull request #1 from rwightman/master ... merge master	5 years ago
Ross Wightman	13cf68850b	Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111	5 years ago
Ross Wightman	56608c9070	Bump version for pypi	5 years ago
Ross Wightman	0ebcd2eca1	No default img size for inference script. Fixes #109	5 years ago
Ross Wightman	1a8f5900ab	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	5 years ago