Method initialize

lib/VIT/decoder/decoder_p.py:111–112

(self)

        return x

    def initialize(self):
        weight_init(self)

class Attention(nn.Module):
    def __init__(self, dim, num_heads, bias,mode):