torchvision/utils.py · f14682a8178650f26544c1d4ac91cc31e0d7f458 · OpenDAS / vision

Generalize ConvNormActivation function to accept tuple for some parameters (#6251) · f14682a8

YosuaMichael authored Jul 08, 2022

* Make ConvNormActivation function accept tuple for kernel_size, stride, padding, and dilation

* Fix the method to get the conv_dim

* Simplify if-elif logic

f14682a8

utils.py 22.5 KB

Replace utils.py