Fix linear 2 substitution index
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
This commit is contained in:
parent
7905f9dd40
commit
987f76840a
1 changed files with 1 additions and 1 deletions
|
@ -46,7 +46,7 @@ def get_tensor_name(name: str) -> str:
|
||||||
if "linear_1" in name:
|
if "linear_1" in name:
|
||||||
name = name.replace("linear_1", "0")
|
name = name.replace("linear_1", "0")
|
||||||
if "linear_2" in name:
|
if "linear_2" in name:
|
||||||
name = name.replace("linear_2", "1")
|
name = name.replace("linear_2", "2")
|
||||||
return name
|
return name
|
||||||
|
|
||||||
if "projection" in name:
|
if "projection" in name:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue