class BertFineTuning(torch.nn.Module):
def __init__(self):
super().__init__()
self.fc_class = torch.nn.Linear(768,num_classes)
def forward(self,x):
last_layer,embedding = bert(x)
embedding = torch.mean(last_layer,dim=1)
out = self.fc(embedding)
return out
I think there may be a typo in the read.me.
Instead of self.fc in the forward function, it should be self.fc_class.
Could the assignment submission also be posted on Sakai?
Thank you!!
I think there may be a typo in the read.me. Instead of self.fc in the forward function, it should be self.fc_class. Could the assignment submission also be posted on Sakai? Thank you!!