empose/nn/models.py [215:226]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        joints_hat = self.maybe_do_fk(pose_hat, shape_hat)

        return {'pose_hat': pose_hat[:, :, 3:],
                'root_ori_hat': pose_hat[:, :, :3],
                'shape_hat': shape_hat,
                'joints_hat': joints_hat}

    def backward(self, batch: ABatch, model_out, writer=None, global_step=None):
        """The backward pass."""
        pose_hat, root_ori_hat, shape_hat = model_out['pose_hat'], model_out['root_ori_hat'], model_out['shape_hat']

        n, f = batch.batch_size, batch.seq_length
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



empose/nn/models.py [316:327]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        joints_hat = self.maybe_do_fk(pose_hat, shape_hat)

        return {'pose_hat': pose_hat[:, :, 3:],
                'root_ori_hat': pose_hat[:, :, :3],
                'shape_hat': shape_hat,
                'joints_hat': joints_hat}

    def backward(self, batch: ABatch, model_out, writer=None, global_step=None):
        """The backward pass."""
        pose_hat, root_ori_hat, shape_hat = model_out['pose_hat'], model_out['root_ori_hat'], model_out['shape_hat']

        n, f = batch.batch_size, batch.seq_length
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -