Commit Graph

5 Commits (a1886346df714b47880cc0f278c792f0a4c36261)

Author SHA1 Message Date
Vyacheslav Shults 44ed0f85e8 Refactor imports and indents according to PEP.
5 years ago
Ross Wightman c60069c1eb Annotate types on drop fns to avoid torchscript error
5 years ago
Ross Wightman f1d5f8a6c4 Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
5 years ago
Ross Wightman f902bcd54c Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
6 years ago
Ross Wightman a99ec4e7d1 A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
6 years ago