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 3a9a40c commit 6296a1fCopy full SHA for 6296a1f
benchmarl/models/cnn.py
@@ -55,8 +55,8 @@ class Cnn(Model):
55
56
The BenchMARL CNN accepts multiple inputs of 2 types:
57
58
- - images: Tensors of shape (*batch, X,Y,C)
59
- - arrays: Tensors of shape (*batch, F)
+ - images: Tensors of shape ``(*batch,X,Y,C)``
+ - arrays: Tensors of shape ``(*batch,F)``
60
61
The CNN model will check that all image inputs have the same shape (excluding the last dimension)
62
and cat them along that dimension before processing them with :class:`torchrl.modules.ConvNet`.
0 commit comments