将模型运行在CNN上时，运行到loss.backward(),模型停止运行 没有任何提示

[FOXaaFOX]

除了主文件的类名外，其他都没有修改，原模型可以运行通过，还麻烦请教，为什么运行到损失向前传播的时候，训练就停止了。
class BertCNN(BertPreTrainedModel):
def init(self, config):
super(BertCNN, self).init(config)
self.num_labels = config.num_labels
self.bert = BertModel(config)
self.dropout = nn.Dropout(config.hidden_dropout_prob)
self.convs = Conv1d(config.hidden_size, n_filters, filter_sizes)
self.classifier = nn.Linear(len(filter_sizes) * n_filters, self.num_labels)
self.init_weights()
def forward(self, input_ids, token_type_ids=None, attention_mask=None, head_mask=None,labels=None):
encoded_layers,_ = self.bert(input_ids, token_type_ids, attention_mask)
encoded_layers = self.dropout(encoded_layers)
encoded_layers = encoded_layers.permute(0, 2, 1)
conved = self.convs(encoded_layers)
pooled = [F.max_pool1d(conv, conv.shape[2]).squeeze(2)
for conv in conved]
cat = self.dropout(torch.cat(pooled, dim=1))
logits = self.classifier(cat)
return logits

[FOXaaFOX]

问题已经解决 应该是pytorch在不同os的问题，我从win10切换到linux 运行正常