kartikdutt18 / mlpack-PyTorch-Weight-Translator

Simple Repository to transfer weights from PyTorch to mlpack.
10 stars 4 forks source link

Added Weight & Bias extractor for nn.ConvTranspose2d #2

Closed Davidportlouis closed 3 years ago

Davidportlouis commented 3 years ago

C++ weight_converter implementation is still pending.

kartikdutt18 commented 3 years ago

Thanks for this!

kartikdutt18 commented 3 years ago

We can open a separate PR for C++ side of things.