Aleksey Zimin
Results
2
comments of
Aleksey Zimin
The issue in my case was due to my machine's CUDA Toolkit having version 10.1 but my pytorch installation CUDA toolkit was 11.3, Installing pytorch with cuda 10.1 solved the...
@FayeXXX just check your pytorch version. if it is >= 1.2.0, do the following change to your init(): modify self.apply(self.init_weights) to self.init_weights() ``` super(BertForMultiLable, self).__init__(config) self.bert = BertModel(config) self.dropout =...