fix len error (#1981)
fix len error (#1981)

Co-authored-by: eellison <32bcb86024d493ca261d5686b38111589a7092c1@fb.com>
pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision