pytorch-revgrad:实现梯度反转层的最小 pytorch 包

时间:2021-05-29 09:59:06
【文件属性】:
文件名称: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)

网友评论