551 Commits (470220b1f4c61ad7deb16dbfb8917089e842cd2a)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	88129b2569	Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.	5 years ago
Ross Wightman	f28170df3f	Fix an an untested change, remove a debug print	5 years ago
Ross Wightman	eb7653614f	Monster commit, activation refactor, VoVNet, norm_act improvements, more ... * refactor activations into basic PyTorch, jit scripted, and memory efficient custom auto * implement hard-mish, better grad for hard-swish * add initial VovNet V1/V2 impl, fix #151 * VovNet and DenseNet first models to use NormAct layers (support BatchNormAct2d, EvoNorm, InplaceIABN) * Wrap IABN for any models that use it * make more models torchscript compatible (DPN, PNasNet, Res2Net, SelecSLS) and add tests	5 years ago
Ross Wightman	ff94ffce61	Merge branch 'master' into densenet_update_and_more	5 years ago
Ross Wightman	5966654052	Update requirements so PyTorch 1.4 is min, add separate sotabench reqs to force 1.4 and prevent error with 1.5	5 years ago
Ross Wightman	0ea53cecc3	Merge branch 'master' into densenet_update_and_more	5 years ago
Ross Wightman	6441e9cc1b	Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.	5 years ago
Ross Wightman	d79ac48626	Update sotabench.py	5 years ago
Ross Wightman	e881383b4e	Merge pull request #154 from rwightman/tests_bugfixes ... Add backward and default_cfg tests and fix a few issues found. Fix #153	5 years ago
Ross Wightman	4d13db538f	Add x48d ResNext101s to test exclude for ubuntu	5 years ago
Ross Wightman	4212cd3b9f	Another attempt at getting Ubuntu test runner to work	5 years ago
Ross Wightman	20329f2630	Bring down test resolutions to see if we can at least do a fwd on the L2 models	5 years ago
Ross Wightman	3873ea710e	Minor test change	5 years ago
Ross Wightman	afb6bd0669	Add backward and default_cfg tests and fix a few issues found. Fix #153	5 years ago
Ross Wightman	ea2e59ca36	Merge pull request #150 from rwightman/regnet ... Add RegNet models and weights	5 years ago
Ross Wightman	50658b9a67	Add RegNet models and weights	5 years ago
Ross Wightman	7df83258c9	Merge branch 'master' into densenet_update_and_more	5 years ago
Ross Wightman	dab9935b36	Merge pull request #148 from rwightman/drop_block_improve ... Improve dropblock impl, add fast variant, better AMP speed, inplace…	5 years ago
Ross Wightman	1904ed8fec	Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups	5 years ago
Ross Wightman	63addb741f	Merge pull request #146 from rwightman/inceptionv3_fix ... Remove annoying torchvision InceptionV3 dependency on scipy and insanely slow tru…	5 years ago
Ross Wightman	17270c69b9	Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.	5 years ago
Ross Wightman	c4ca016656	Merge pull request #145 from rwightman/resnest ... ResNeSt	5 years ago
Ross Wightman	208e7912f7	Missed one of the abalation model entrypoints, update README	5 years ago
Ross Wightman	5bd1ad13e7	Refactor test indent	5 years ago
Ross Wightman	9cc289f18c	Exclude EfficientNet-L2 models from test	5 years ago
Ross Wightman	2f884a0ce5	Add resnest14, resnest26, and two of the abalation grouped resnest50 models	5 years ago
Ross Wightman	f4cdc2ac31	Add ResNeSt models	5 years ago
Ross Wightman	780860d140	Add norm_act factory method, move JIT of norm layers to factory	5 years ago
Ross Wightman	14edacdf9a	DenseNet converted to support ABN (norm + act) modules. Experimenting with EvoNorm, IABN	5 years ago
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