【文件属性】:
文件名称:pytorch-revgrad:实现梯度反转层的最小 pytorch 包
文件大小:10KB
文件格式:ZIP
更新时间:2021-05-29 09:59:06
pytorch domain-adaptation gradient-reversal Python
pytorch-revgrad
这个包为 pytorch 模块实现了一个梯度反转层。
示例用法
import torch
from pytorch_revgrad import RevGrad
model = torch . nn . Sequential (
torch . nn . Linear ( 10 , 5 ),
torch . nn . Linear ( 5 , 2 ),
RevGrad ()
)
【文件预览】:
pytorch-revgrad-master
----.gitignore(228B)
----pyproject.toml(555B)
----src()
--------pytorch_revgrad()
----LICENSE.md(1KB)
----.travis.yml(209B)
----LICENSE(1KB)
----README.md(1003B)
----tests()
--------test_layer.py(1KB)
----docs()
--------index.md(656B)
--------Makefile(618B)
--------conf.py(6KB)
--------k8s-pod.yaml(2KB)