(self)
| 488 | |
| 489 | class BLMLP(nn.Module): |
| 490 | def __init__(self): |
| 491 | super(BLMLP, self).__init__() |
| 492 | self.W = nn.Parameter(nn.init.xavier_uniform_(torch.empty(args.student_embed_size, args.student_embed_size))) |
| 493 | self.act = nn.LeakyReLU(negative_slope=0.5) |
| 494 | |
| 495 | def forward(self, embeds): |
| 496 | pass |