Skip to content

Commit 91c7554

Browse files
authored
Fixed a small typo in the cnn.py (#153)
Change weight initialization statement from Xavier to Kaiming
1 parent 845d368 commit 91c7554

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

rsl_rl/networks/cnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def output_dim(self) -> tuple[int, int] | int:
147147
return self._output_dim
148148

149149
def init_weights(self) -> None:
150-
"""Initialize the weights of the CNN with Xavier initialization."""
150+
"""Initialize the weights of the CNN with Kaiming initialization."""
151151
for idx, module in enumerate(self):
152152
if isinstance(module, nn.Conv2d):
153153
torch.nn.init.kaiming_normal_(module.weight)

0 commit comments

Comments
 (0)