Update conv2d_layers.py

[instinct79]

Resolve container_abcs not found in torch._six, for pyTorch 1.8

[maatthc]

I can confirm that this solves this issue:

from geffnet import efficientnet_b0 File "/home/ec2-user/.cache/torch/hub/rwightman_gen-efficientnet-pytorch_master/geffnet/__init__.py", line 1, in <module> from .gen_efficientnet import * File "/home/ec2-user/.cache/torch/hub/rwightman_gen-efficientnet-pytorch_master/geffnet/gen_efficientnet.py", line 33, in <module> from .conv2d_layers import select_conv2d File "/home/ec2-user/.cache/torch/hub/rwightman_gen-efficientnet-pytorch_master/geffnet/conv2d_layers.py", line 11, in <module> from torch._six import container_abcs ImportError: cannot import name 'container_abcs' from 'torch._six' (/usr/local/lib64/python3.7/site-packages/torch/_six.py)

[wagonduke]

Could this be merged? Works for me 👍

[hectorLop]

It works for me too

[hannadiamond]

@rwightman please merge to fix the issue, this fixed it for me