Ожидается более 1 значения на канал при обучении, получен входной размер torch.Size ([1, xx]) - PullRequest
Новая
       83

Ожидается более 1 значения на канал при обучении, получен входной размер torch.Size ([1, xx])

0 голосов
/ 23 апреля 2020

Рассмотрим следующий сетевой фрагмент: 

def __init__(self, model, n_class, dropout_rate,device):
    super(NewModel, self).__init__()

    self.bert = model
    self.linear = nn.Linear(self.bert.config.hidden_size, 2)
    self.linear_1 = nn.Linear(self.bert.config.hidden_size, self.bert.config.hidden_size)

    self.dropout_rate = dropout_rate
    self.dropout_1 = nn.Dropout(p = self.dropout_rate)

    self.activation = nn.LeakyReLU()

    self.bn = nn.BatchNorm1d(num_features = self.bert.config.hidden_size)

def forward(self, batch):
    outputs = self.bert(
        input_ids = batch[0].to(self.device),
        attention_mask = batch[1].to(self.device),
        token_type_ids = None,
        position_ids = None,
        head_mask = None,
        inputs_embeds = None,
    )

    output = outputs[0]
    pooled_output=output[:,0]
    pooled_output = pooled_output.unsqueeze(0)

    pooled_output_1 = self.dropout_1(self.bn(pooled_output))

    logits = self.linear(F.leaky_relu(self.linear_1(pooled_output_1)))

Размер моего пакета равен 16, и во время обучения я получаю эту ошибку: 

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 16])

Я уже установил drop_last=True в DataLoader, но ошибка сохраняется.

Любая помощь будет принята с благодарностью.

1 Ответ

0 голосов
/ 25 апреля 2020

Размеры ваших тензоров неверны: размер партии всегда должен стоять первым (а ошибка was got input size torch.Size([1, 16])). Загрузчик данных pytorch загружает данные в следующей форме: (batch_size, input_tensor_size). Следовательно, input_ids подается с первым элементом вашей партии, а attention_mask - со вторым, что, я полагаю, не то, что вы хотели (за исключением того, что вы изменили тензор, выводимый загрузчиком данных, который будет источником ошибка)

...