Function build_position_encoding

models/apr/pencoder.py:72–78

(config)

def build_position_encoding(config):
    N_steps = config.hidden_dim // 2
    if config.learn_embedding_with_pose_token:
        position_embedding = PositionEmbeddingLearnedWithPoseToken(N_steps, config.device)
    else:
        position_embedding = PositionEmbeddingLearned(N_steps, config.device)
    return position_embedding