From fea7acecc5529be495da05b2bfca93f6f20182c1 Mon Sep 17 00:00:00 2001 From: My Name Date: Wed, 3 Jul 2024 11:20:52 -0700 Subject: [PATCH] debugged DNN.init() to correctly accept 4 arguments when it was previously missing a 'max_control_cost' argument --- src/dynadojo/baselines/dnn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/dynadojo/baselines/dnn.py b/src/dynadojo/baselines/dnn.py index e620bae..8438fd6 100644 --- a/src/dynadojo/baselines/dnn.py +++ b/src/dynadojo/baselines/dnn.py @@ -170,8 +170,9 @@ class DNN(TorchBaseClass): def __init__(self, embed_dim, timesteps, + max_control_cost, **kwargs): - super().__init__(embed_dim, timesteps, **kwargs) + super().__init__(embed_dim, timesteps, max_control_cost, **kwargs) self.model = torch.nn.Sequential( torch.nn.Linear(self.embed_dim, embed_dim*10), torch.nn.ReLU(),