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 5080235 commit f9b5840Copy full SHA for f9b5840
intermediate_source/torchvision_tutorial.rst
@@ -263,7 +263,7 @@ way of doing it:
263
# be [0]. More generally, the backbone should return an
264
# OrderedDict[Tensor], and in featmap_names you can choose which
265
# feature maps to use.
266
- roi_pooler = torchvision.ops.MultiScaleRoIAlign(featmap_names=[0],
+ roi_pooler = torchvision.ops.MultiScaleRoIAlign(featmap_names=['0'],
267
output_size=7,
268
sampling_ratio=2)
269
0 commit comments