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()
)-
Notifications
You must be signed in to change notification settings - Fork 14
A minimal pytorch package implementing a gradient reversal layer.
License
MIT, MIT licenses found
Licenses found
MIT
LICENSE
MIT
LICENSE.md
janfreyberg/pytorch-revgrad
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
A minimal pytorch package implementing a gradient reversal layer.
Topics
Resources
License
MIT, MIT licenses found
Licenses found
MIT
LICENSE
MIT
LICENSE.md
Stars
Watchers
Forks
Packages 0
No packages published