AI Dynamics

Global AI News Aggregator

Keras Core now supports arbitrary PyTorch Modules with TorchModuleWrapper

Now, it's possible to use arbitrary `Modules` inside a Keras Core model/layer (with the torch backend). The `TorchModuleWrapper` class turns them into Keras layers, keeping tracking of trainable weights / etc.

→ View original post on X — @fchollet,

Commentaires

Leave a Reply

Your email address will not be published. Required fields are marked *