почему эта штука не так в ноутбуке Jupyter, а в терминале? - PullRequest
0 голосов
/ 04 июня 2019

Когда я запустил этот кусок кода в jupyter, он не работал правильно.Ошибка:

TypeError                                 Traceback (most recent call
last) <ipython-input-16-76d9b57527d4> in <module>
     14 embed_layer = emb()
     15 
---> 16 inp = torch.tensor([ 1, 2, 3, 4, 5, 6 ])
     17 
     18 out = embed_layer( inp )

TypeError: 'module' object is not callable

НО, когда я запустил его в терминале, я могу запустить его правильно.Я пытаюсь обучить модель для встраивания слов.

import torch
import torch.nn as nn

class emb( nn.Module ):

    def __init__( self ):
        super( emb, self ).__init__()
        self.embed_layer = nn.Embedding(8,3)

    def forward( self, inp ):
        out = self.embed_layer( inp )
        return out

embed_layer = emb()

inp = torch.tensor([ 1, 2, 3, 4, 5, 6 ])

out = embed_layer( inp )
print(out)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...