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 845d368 commit 91c7554Copy full SHA for 91c7554
rsl_rl/networks/cnn.py
@@ -147,7 +147,7 @@ def output_dim(self) -> tuple[int, int] | int:
147
return self._output_dim
148
149
def init_weights(self) -> None:
150
- """Initialize the weights of the CNN with Xavier initialization."""
+ """Initialize the weights of the CNN with Kaiming initialization."""
151
for idx, module in enumerate(self):
152
if isinstance(module, nn.Conv2d):
153
torch.nn.init.kaiming_normal_(module.weight)
0 commit comments