From 6296a1f3f45f3266e0b9fea9306b7c571234c674 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Wed, 17 Apr 2024 10:12:42 +0100 Subject: [PATCH] [Docs] Fix typo --- benchmarl/models/cnn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarl/models/cnn.py b/benchmarl/models/cnn.py index a6956fe3..afce12c5 100644 --- a/benchmarl/models/cnn.py +++ b/benchmarl/models/cnn.py @@ -55,8 +55,8 @@ class Cnn(Model): The BenchMARL CNN accepts multiple inputs of 2 types: - - images: Tensors of shape (*batch, X,Y,C) - - arrays: Tensors of shape (*batch, F) + - images: Tensors of shape ``(*batch,X,Y,C)`` + - arrays: Tensors of shape ``(*batch,F)`` The CNN model will check that all image inputs have the same shape (excluding the last dimension) and cat them along that dimension before processing them with :class:`torchrl.modules.ConvNet`.