From 90a74efd89a64861aa684759edf105a39dba2f39 Mon Sep 17 00:00:00 2001 From: iperov Date: Sat, 26 Jun 2021 23:06:30 +0400 Subject: [PATCH] fix --- models/Model_AMP/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_AMP/Model.py b/models/Model_AMP/Model.py index 1588f56..2c60741 100644 --- a/models/Model_AMP/Model.py +++ b/models/Model_AMP/Model.py @@ -207,7 +207,7 @@ class AMPModel(ModelBase): class Inter(nn.ModelBase): def on_build(self): - self.dense2 = nn.Dense( ae_dims, lowest_dense_res * lowest_dense_res * ae_dims ) + self.dense2 = nn.Dense( ae_dims, lowest_dense_res * lowest_dense_res * inter_dims ) def forward(self, inp): x = inp