By popular demand, we've automated the "turn the Module into a Keras Layer" step. So now you can just… use torch Modules in a Keras model/layer. No extra step needed. Just like you can use Keras models/layers as Modules.
Automated PyTorch Module Integration with Keras Layers
By
–
Leave a Reply