Supported Layers Tracker
Closed this issue · 0 comments
mohmdelsayed commented
Here is a list of the torch.nn modules to be supported for HesScale.
- Loss functions
- MSELoss
- CrossEntropyLoss
- Activations
- ReLU
- Sigmoid
- Tanh
- LeakyReLU
- LogSigmoid
- ELU
- SELU
- Others:
- Linear
- Dropout
- Flatten
- CNN related layers:
- MaxPool1d
- MaxPool2d
- AvgPool1d
- MaxPool3d
- AvgPool2d
- AvgPool3d
- ZeroPad2d
- Conv1d
- Conv2d
- Conv3d
- ConvTranspose1d
- ConvTranspose2d
- ConvTranspose3d
- Recurrent related layers:
- RNN
- LSTM