patchify 和 unpatchify

你好，只是一个建议，也许可以在 patchify 和 unpatchify 中将魔术数字“3”改为 in_chans，这样模型可以更好地适应灰度图像。