Selected training excerpt · lines 415–415 · source SHA-256 5718458b537d
415 self.mask_token = nn.Parameter(torch.zeros(1, 1, self.d_pred))