search

MIC-DKFZ / dynamic-network-architectures

Apache License 2.0
66 stars 27 forks source link

support subclasses of nn.ConvX #7

Open elazarcoh opened 2 weeks ago

elazarcoh commented 2 weeks ago

Allows me to use customized conv_op, e.g. from https://github.com/NVIDIA/partialconv/blob/master/models/partialconv3d.py