3 Nb-@s(ddlZddljZGdddejZdS)Ncs$eZdZfddZddZZS)DepthWiseSeperableConvcsLtjd|kr|d=tj||f|d|i||_tj||dd|_dS)Ngroups) kernel_size)super__init__nnConv2d depthwise pointwise)selfZin_dimZout_dimargskwargs) __class__S/home/ec2-user/SageMaker/lama/saicinpainting/training/modules/depthwise_sep_conv.pyrs  zDepthWiseSeperableConv.__init__cCs|j|}|j|}|S)N)r r )r xoutrrrforwards  zDepthWiseSeperableConv.forward)__name__ __module__ __qualname__rr __classcell__rr)rrrs r)torchtorch.nnrModulerrrrrs