Unverified Commit e0a64f95 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Fix Retiarii version issues on full test (#4371)

parent f4150039
...@@ -5,7 +5,7 @@ tensorflow ...@@ -5,7 +5,7 @@ tensorflow
keras == 2.4.3 keras == 2.4.3
torch == 1.10.0+cu111 torch == 1.10.0+cu111
torchvision == 0.11.1+cu111 torchvision == 0.11.1+cu111
pytorch-lightning >= 1.4.2 pytorch-lightning >= 1.5.0
onnx onnx
peewee peewee
graphviz graphviz
......
...@@ -32,7 +32,7 @@ class ENASLayer(mutables.MutableScope): ...@@ -32,7 +32,7 @@ class ENASLayer(mutables.MutableScope):
if self.skipconnect is not None: if self.skipconnect is not None:
connection = self.skipconnect(prev_layers[:-1]) connection = self.skipconnect(prev_layers[:-1])
if connection is not None: if connection is not None:
out += connection out = out + connection
return self.batch_norm(out) return self.batch_norm(out)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment