How to specify the batch_size in PyTorch Sequantial model?

前端 未结 0 922
花落未央
花落未央 2020-11-30 22:09

I have a Sequential model in PyTorch:

model = nn.Sequential(
        nn.Embedding(alphabet_size, 64),
        nn.LSTM(64, stateful=False, return_sequences=Tru         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题