An insignificant comment error

I am a student and recently I was studying this project and I found a very obscure comment error. In the unpatchify function in models.py, imgs is annotated as (N,H,W,C), but it should be (N,C,H,W)