Imported from https://github.com/CHUNYUWANG/imu-human-pose-pytorch