Treinamento do Modelo

Visualizando 2 posts - 1 até 2 (de 2 do total)
  • Autor
    Posts
  • #28277
    Felipe
    Participante

    Na parte 1 – Redes neurais artificiais: Classificação binária – base breast cancer na etapa de Treinamento do Modelo, fiz o seguinte código, conforme proposto na aula:

    for epoch in range(100):
    running_loss = 0.

    for data in train_loader:
    inputs, labels = data
    optimizer.zero_grad()

    outputs = classificador(inputs)

     

    Entretanto, acontece o seguinte erro no google collab:

    —————————————————————————

    RuntimeError Traceback (most recent call last)

    <ipython-input-58-cb716bf3324b> in <module>()
    6 optimizer.zero_grad()
    7
    —-> 8 outputs = classificador.forward(inputs)
    9 print(outputs)
    10

    3 frames
    /usr/local/lib/python3.7/dist-packages/torch/nn/functional.py in linear(input, weight, bias)
    1368 if input.dim() == 2 and bias is not None:
    1369 # fused op is marginally faster
    -> 1370 ret = torch.addmm(bias, input, weight.t())
    1371 else:
    1372 output = input.matmul(weight.t())

    RuntimeError: size mismatch, m1: [10 x 16], m2: [30 x 16] at /pytorch/aten/src/TH/generic/THTensorMath.cpp:136

    Como eu poderia fazer pra resolver?

    #28279
    Profile photo ofdennyDenny Ceccon
    Moderador

    Olá Felipe,

    Deve haver algum erro na criação da rede, pois a mensagem de erro aponta incompatibilidade no tamanho das camadas ocultas.

    Verifique o código correspondente e, se continuar tendo erro, por favor reproduza seu código aqui.

Visualizando 2 posts - 1 até 2 (de 2 do total)
  • Você deve fazer login para responder a este tópico.