A pytorch module (and function) to reverse gradients.
pip install pytorch_revgrad==0.0.1
pytorch-revgrad
This package implements a gradient reversal layer for pytorch modules.
import torch
from pytorch_revgrad import RevGrad
model = torch.nn.Sequential(
torch.nn.Linear(10, 5),
torch.nn.Linear(5, 2),
RevGrad()
)
The Tidelift Subscription provides access to a continuously curated stream of human-researched and maintainer-verified data on open source packages and their licenses, releases, vulnerabilities, and development practices.
Learn more →
Something wrong with this page? Make a suggestion
Export .ABOUT file for this package
Last synced: 2023-06-02 20:48:49 UTC
Login to resync this project