Spaces:

riiswa
/

RL-Interpretable-Policy-via-Kolmogorov-Arnold-Network

Running

App Files Files Community

riiswa commited on May 5

Commit

3735566

•

1 Parent(s): a364ed7

debugging

Files changed (1) hide show

interpretable.py +1 -1

interpretable.py CHANGED Viewed

@@ -34,7 +34,7 @@ class InterpretablePolicyExtractor:
         if dataset["train_label"].ndim == 1 and not self._action_is_discrete:
             dataset["test_label"] = dataset["test_label"][:, None]
         for k,v in dataset.items():
-            print(k, v.shape)
         return self.policy.train(dataset, opt="LBFGS", steps=steps, loss_fn=self.loss_fn)
     def forward(self, observation):

         if dataset["train_label"].ndim == 1 and not self._action_is_discrete:
             dataset["test_label"] = dataset["test_label"][:, None]
         for k,v in dataset.items():
+            print(k, v.shape, v.dtype)
         return self.policy.train(dataset, opt="LBFGS", steps=steps, loss_fn=self.loss_fn)
     def forward(self, observation):