We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0cc749d commit 8e4cb26Copy full SHA for 8e4cb26
autointent/modules/scoring/_rnn.py
@@ -203,6 +203,15 @@ def device(self) -> str:
203
"""Get device used for model computations."""
204
return self._device
205
206
+ @device.setter
207
+ def device(self, value: str) -> None:
208
+ """Set device for model computations."""
209
+ self._device = value
210
+
211
+ def get_implicit_initialization_params(self) -> dict[str, Any]:
212
+ """Return default params used in ``__init__`` method."""
213
+ return {"rnn_config": self.rnn_config.model_dump()}
214
215
class SupervisedRNNClassifier(nn.Module):
216
def __init__(
217
self,
0 commit comments