Method __init__

detrsmpl/core/post_processing/speed_up/deciwatch.py:198–229

(self,
                 input_dim=24 * 6,
                 sample_interval=10,
                 encoder_hidden_dim=16,
                 decoder_hidden_dim=16,
                 dropout=0.1,
                 nheads=4,
                 dim_feedforward=256,
                 enc_layers=3,
                 dec_layers=3,
                 activation='leaky_relu',
                 pre_norm=False)

    """

    def __init__(self,
                 input_dim=24 * 6,
                 sample_interval=10,
                 encoder_hidden_dim=16,
                 decoder_hidden_dim=16,
                 dropout=0.1,
                 nheads=4,
                 dim_feedforward=256,
                 enc_layers=3,
                 dec_layers=3,
                 activation='leaky_relu',
                 pre_norm=False):
        super(DeciWatch, self).__init__()
        self.pos_embed_dim = encoder_hidden_dim
        self.pos_embed = self.build_position_encoding(self.pos_embed_dim)

        self.sample_interval = sample_interval

        self.deciwatch_par = {
            'input_dim': input_dim,
            'encoder_hidden_dim': encoder_hidden_dim,
            'decoder_hidden_dim': decoder_hidden_dim,
            'dropout': dropout,
            'nheads': nheads,
            'dim_feedforward': dim_feedforward,
            'enc_layers': enc_layers,
            'dec_layers': dec_layers,
            'activation': activation,
            'pre_norm': pre_norm
        }

        self.transformer = build_model(self.deciwatch_par)

    def build_position_encoding(self, pos_embed_dim):
        N_steps = pos_embed_dim // 2