A pytorch module (and function) to reverse gradients.

Project description

`pytorch-revgrad`

python version

This package implements a gradient reversal layer for pytorch modules.

Example usage

import torch

from pytorch_revgrad import RevGrad

model = torch.nn.Sequential(
    torch.nn.Linear(10, 5),
    torch.nn.Linear(5, 2),
    RevGrad()
)

Project details

Release history Release notifications | RSS feed

This version

0.2.0

Jan 9, 2021

0.1.1

Jan 31, 2020

0.0.1

Mar 30, 2019

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pytorch_revgrad-0.2.0.tar.gz (7.1 kB view hashes)

Uploaded Jan 9, 2021 Source

Built Distribution

pytorch_revgrad-0.2.0-py3-none-any.whl (4.6 kB view hashes)

Uploaded Jan 9, 2021 Python 3

Hashes for pytorch_revgrad-0.2.0.tar.gz

Hashes for pytorch_revgrad-0.2.0.tar.gz
Algorithm	Hash digest
SHA256	`9cf097a7d18cbadddeaec9fef74b258d70b6cb8d0c77f524baab18bffc7d7be9`
MD5	`3ea51b11a52967d8e261d328dcb24932`
BLAKE2b-256	`90c3aecc90deea743b93757e05e235f70eef479509eb3eb11b19f3aa0d528541`

Hashes for pytorch_revgrad-0.2.0-py3-none-any.whl

Hashes for pytorch_revgrad-0.2.0-py3-none-any.whl
Algorithm	Hash digest
SHA256	`2276fb189b2ce26f756a97effe2a6bcf8f7fdc60542c5dfb45c53f09ef123aa7`
MD5	`0cca2daedc6b24de9a6063d2be189b31`
BLAKE2b-256	`ece910b11b186b99c40213dca68cf6c38051b6704a74e1056d3f3ca4c12f14b9`