jgvycjxyfcjyhcjy

好像不是batch size的问题，设1都不行

这个问题是因为 s1_train.py 里函数on_train_epoch_end(self, trainer, pl_module)中的 if 语句未执行，故checkpoint无法保存，直接更改为 if Ture：可以解决。但这么解决似乎不太合理。 这里的 self._should_save_on_train_epoch_end(trainer) 和 self._should_skip_saving_checkpoint(trainer) 应该如何读入呢？

I wonder if the dimensions of these three variables(pose_body, pose_hand, root_orient) are true.