Skip to content

Commit 186ab4a

Browse files
committed
Remove useless attribute depth for mlp network.
1 parent d37d09e commit 186ab4a

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

qmb/crossmlp.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@ def __init__(self, dim_input: int, dim_output: int, hidden_size: tuple[int, ...]
4545
self.dim_input: int = dim_input
4646
self.dim_output: int = dim_output
4747
self.hidden_size: tuple[int, ...] = hidden_size
48-
self.depth: int = len(hidden_size)
4948

5049
dimensions: list[int] = [dim_input] + list(hidden_size) + [dim_output]
5150
linears: list[torch.nn.Module] = [select_linear_layer(i, j) for i, j in zip(dimensions[:-1], dimensions[1:])]

qmb/mlp.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,6 @@ def __init__(self, dim_input: int, dim_output: int, hidden_size: tuple[int, ...]
4444
self.dim_input: int = dim_input
4545
self.dim_output: int = dim_output
4646
self.hidden_size: tuple[int, ...] = hidden_size
47-
self.depth: int = len(hidden_size)
4847

4948
dimensions: list[int] = [dim_input] + list(hidden_size) + [dim_output]
5049
linears: list[torch.nn.Module] = [select_linear_layer(i, j) for i, j in zip(dimensions[:-1], dimensions[1:])]

0 commit comments

Comments
 (0)