DouZero
/
Douzero_Resnet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复参数问题

This commit is contained in:
zhiyang7 2021-12-21 18:05:52 +08:00
parent 42066ae7a9
commit 42f28b1aa9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
douzero/dmc/models.py
View File

@ -128,7 +128,7 @@ class LandlordLstmModelLegacy(nn.Module):
return { return {
'args': ( 'args': (
torch.randn(1, 5, 432, requires_grad=True, device=device), 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'], 'input_names': ['z_batch','x_batch'],
'output_names': ['values'], 'output_names': ['values'],
@ -177,7 +177,7 @@ class FarmerLstmModelLegacy(nn.Module):
return { return {
'args': ( 'args': (
torch.randn(1, 5, 432, requires_grad=True, device=device), 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'], 'input_names': ['z_batch','x_batch'],
'output_names': ['values'], 'output_names': ['values'],