snt.AbstractModule in dm-sonnet=2.0.0
snt.AbstractModule in dm-sonnet=2.0.0
提问于 2020-10-12 19:37:03
我一直在尝试运行2017年编写的代码，它使用tensorflow=1.3.0和dm-sonnet=1.14。许多必需的包已经变得不可用，我被依赖性问题淹没了，所以我决定用新版本的tensorflow重写代码。
我有：
tensorflow=2.3.1
tensorflow-addons=0.11.2
tensorflow-estimator=2.3.0
tensorflow-probability=0.11.1
dm-sonnet=2.0.0许多更新只是简单地将tf1更新到tf2，但我在dm-sonnet版本上遇到了无数的问题。
代码有：
class CosineWeights(snt.AbstractModule):
我得到了：
AttributeError: module 'sonnet' has no attribute 'AbstractModule'
我怀疑这只是一个将sonnet=1.14的代码更新为十四行诗2.0的问题，但我找不到它的“翻译”，或者AbstractModule现在是如何使用的。换用哪个模块，只做snt.Module就足够了吗？如果有任何帮助，我将不胜感激！
回答 1
Stack Overflow用户
发布于 2021-08-26 15:30:35
引用https://github.com/deepmind/sonnet的话它解决了我的问题。希望这能帮上忙~
构建您自己的模块Sonnet强烈鼓励用户子类化snt.模块来定义自己的模块。让我们从创建一个名为MyLinear的简单线性层开始：
class MyLinear(snt.Module):
def __init__(self, output_size, name=None):
super(MyLinear, self).__init__(name=name)
self.output_size = output_size页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/64317125
复制相关文章
相似问题
腾讯云开发者
