File size: 230 Bytes
cc9c7ee |
1 2 3 4 5 6 7 |
import torch.nn as nn
from src.utils.mapper import configmapper
configmapper.map("activations", "relu")(nn.ReLU)
configmapper.map("activations", "logsoftmax")(nn.LogSoftmax)
configmapper.map("activations", "softmax")(nn.Softmax)
|