ding.torch_utils.network

ding.torch_utils.network

Full Source Code

../ding/torch_utils/network/__init__.py

1from .activation import build_activation, Swish 2from .res_block import ResBlock, ResFCBlock 3from .nn_module import fc_block, conv2d_block, one_hot, deconv2d_block, BilinearUpsample, NearestUpsample, \ 4 binary_encode, NoiseLinearLayer, noise_block, MLP, Flatten, normed_linear, normed_conv2d, conv1d_block 5from .normalization import build_normalization 6from .rnn import get_lstm, sequence_mask 7from .soft_argmax import SoftArgmax 8from .transformer import Transformer, ScaledDotProductAttention 9from .scatter_connection import ScatterConnection 10from .resnet import resnet18, ResNet 11from .gumbel_softmax import GumbelSoftmax 12from .gtrxl import GTrXL, GRUGatingUnit 13from .popart import PopArt 14#from .dreamer import Conv2dSame, DreamerLayerNorm, ActionHead, DenseHead 15from .merge import GatingType, SumMerge, VectorMerge