hello，感谢你们提供的代码。 不过，我怀疑你们数据处理时，RandomResizedCrop把W和H传反了，能麻烦你们看看吗？ 另外请问为啥ratio要设为(0.9, 1.0)？这不是会导致偏胖吗？ 另外请问为啥dwpose最后不resize回原图的分辨率？这不是会导致骨架和原图对不上么？ 最后请问这套代码能跑在你们摩尔线程的卡上吗？我用80g的a100跑的，512*768的分辨率都会爆显存，你们摩尔线程要什么卡才能跑起来？
Activity