diff --git a/douzero/dmc/models.py b/douzero/dmc/models.py index 16f8b50..c876152 100644 --- a/douzero/dmc/models.py +++ b/douzero/dmc/models.py @@ -448,7 +448,7 @@ class General_Model: action = torch.randint(values.shape[0], (1,))[0] else: action = torch.argmax(values,dim=0)[0] - return dict(action=action, max_value=torch.max(values)) + return dict(action=action) def share_memory(self): self.models['landlord'].share_memory() @@ -516,7 +516,7 @@ class OldModel: action = torch.randint(values.shape[0], (1,))[0] else: action = torch.argmax(values,dim=0)[0] - return dict(action=action, max_value=torch.max(values)) + return dict(action=action) def share_memory(self): self.models['landlord'].share_memory() @@ -587,7 +587,7 @@ class Model: action = torch.randint(values.shape[0], (1,))[0] else: action = torch.argmax(values,dim=0)[0] - return dict(action=action, max_value=torch.max(values)) + return dict(action=action) def share_memory(self): self.models['landlord'].share_memory()