From 42f28b1aa9eb17734a66478f0e27a6b3a6e638b4 Mon Sep 17 00:00:00 2001 From: zhiyang7 Date: Tue, 21 Dec 2021 18:05:52 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8F=82=E6=95=B0=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- douzero/dmc/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/douzero/dmc/models.py b/douzero/dmc/models.py index a92727e..1aab661 100644 --- a/douzero/dmc/models.py +++ b/douzero/dmc/models.py @@ -128,7 +128,7 @@ class LandlordLstmModelLegacy(nn.Module): return { 'args': ( torch.randn(1, 5, 432, requires_grad=True, device=device), - torch.randn(1, 887, requires_grad=True, device=device), + torch.randn(1, 860, requires_grad=True, device=device), ), 'input_names': ['z_batch','x_batch'], 'output_names': ['values'], @@ -177,7 +177,7 @@ class FarmerLstmModelLegacy(nn.Module): return { 'args': ( torch.randn(1, 5, 432, requires_grad=True, device=device), - torch.randn(1, 1219, requires_grad=True, device=device), + torch.randn(1, 1192, requires_grad=True, device=device), ), 'input_names': ['z_batch','x_batch'], 'output_names': ['values'],