提交 4f8552aa 编写于 作者: W weishengyu

dbg

上级 8211880d
...@@ -14,7 +14,7 @@ class Identity(nn.Layer): ...@@ -14,7 +14,7 @@ class Identity(nn.Layer):
class TheseusLayer(nn.Layer): class TheseusLayer(nn.Layer):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(TheseusLayer, self).__init__() super(TheseusLayer, self).__init__()
self.res_dict = None self.res_dict = {}
# stop doesn't work when stop layer has a parallel branch. # stop doesn't work when stop layer has a parallel branch.
def stop_after(self, stop_layer_name: str): def stop_after(self, stop_layer_name: str):
......
...@@ -154,7 +154,7 @@ class VGGNet(TheseusLayer): ...@@ -154,7 +154,7 @@ class VGGNet(TheseusLayer):
x = self.drop(x) x = self.drop(x)
x = self.fc3(x) x = self.fc3(x)
if self.res_dict: if self.res_dict:
for res_key in self.res_dict: for res_key in list(self.res_dict):
res_dict[res_key] = self.res_dict.pop(res_key) res_dict[res_key] = self.res_dict.pop(res_key)
return x return x
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册