Responder a: Duvidas sobre o Tensorflow Hub

Home Fóruns Fórum Deep Learning com TensorFlow Hub Duvidas sobre o Tensorflow Hub Responder a: Duvidas sobre o Tensorflow Hub

#45722
Denny Ceccon
Moderador

    Olá Luiz,

    É possível, mas não é muito simples. Os modelos têm tantos neurônios na camada de saída quantas classes originais, ou seja, para adicionar ou excluir movimentos, você precisa adicionar ou remover neurônios (e suas conexões), mas não dá pra fazer isso sem desestruturar todo o conhecimento da rede.

    Uma possibilidade é trocar a última camada inteira e treinar somente esta camada com dados rotulados contendo suas classes de interesse, esta técnica se chama transfer learning. Se você quiser dar uma olhada como funciona, eu lembro que falamos de transfer learning em nosso curso de PyTorch.

    • Esta resposta foi modificada 2 meses, 1 semana atrás por Denny Ceccon.