python - How to write a pytorch sequential model? -


so far wrote mlp,rnn , cnn in keras+tf pytorch gaining popularity inside deep learning communities , started learn framework. biggest fan of sequential models of keras allow make simple models fast. seen pytorch has functionality dont know how code one. tried way

import torch import torch.nn nn  net = nn.sequential() net.add(nn.linear(3, 4)) net.add(nn.sigmoid()) net.add(nn.linear(4, 1)) net.add(nn.sigmoid()) net.float()  print(net) 

but giving error.

attributeerror: 'sequential' object has no attribute 'add'

also if possible can give simple examples rnn , cnn models in pytorch sequenctial model.

sequential not have add method @ moment, though there debate adding functionality.

as can read in documentation nn.sequential takes argument layers separeted sequence of arguments or ordereddict.

if have model lots of layers, can create list first , use * operator expand list positional arguments, this:

layers = [] layers.append(nn.linear(3, 4)) layers.append(nn.sigmoid()) layers.append(nn.linear(4, 1)) layers.append(nn.sigmoid())  net = nn.sequential(*layers) 

this result in similar structure of code, adding directly.


Comments

Popular posts from this blog

resizing Telegram inline keyboard -

command line - How can a Python program background itself? -

php - "cURL error 28: Resolving timed out" on Wordpress on Azure App Service on Linux -