mikeizbicki / cmc-csci181-deeplearning

deep learning course materials
15 stars 6 forks source link

typo and hw submission #39

Closed AlexKer closed 4 years ago

AlexKer commented 4 years ago
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!!

mikeizbicki commented 4 years ago

Both changes are fixed.