示例#1
0
from vedastr.utils import Registry

TRANSFORMER_ENCODER_LAYERS = Registry('transformer_encoder_layer')
TRANSFORMER_DECODER_LAYERS = Registry('transformer_decoder_layer')
示例#2
0
from vedastr.utils import Registry

DECODERS = Registry('decoder')
示例#3
0
from vedastr.utils import Registry

SEQUENCE_DECODERS = Registry('sequence_decoder')
示例#4
0
from vedastr.utils import Registry

TRANSFORMS = Registry('transforms')
示例#5
0
from vedastr.utils import Registry

BRICKS = Registry('brick')
示例#6
0
from vedastr.utils import Registry

CONVERTERS = Registry('convert')
示例#7
0
from vedastr.utils import Registry

DATASETS = Registry('dataset')
示例#8
0
from vedastr.utils import Registry

SAMPLER = Registry('sampler')
示例#9
0
from vedastr.utils import Registry

TRANSFORMER_ATTENTIONS = Registry('transformer_attention')
示例#10
0
from vedastr.utils import Registry

BACKBONES = Registry('backbone')
示例#11
0
from vedastr.utils import Registry

SEQUENCE_ENCODERS = Registry('sequence_encoder')
示例#12
0
from vedastr.utils import Registry

POSITION_ENCODERS = Registry('position_encoder')
示例#13
0
from vedastr.utils import Registry

DATALOADERS = Registry('dataloader')
示例#14
0
from vedastr.utils import Registry

TRANSFORMER_FEEDFORWARDS = Registry('transformer_feedforward')
示例#15
0
from vedastr.utils import Registry

UTILS = Registry('utils')
示例#16
0
import torch.nn as nn

from vedastr.utils import Registry

CRITERIA = Registry('criterion')

# CTCLoss = nn.CTCLoss
# CRITERIA.register_module(CTCLoss)
#
#
# CrossEntropyLoss = nn.CrossEntropyLoss
# CRITERIA.register_module(CrossEntropyLoss)
示例#17
0
from torch.optim import lr_scheduler

from vedastr.utils import Registry


LR_SCHEDULERS = Registry('lr_scheduler')
示例#18
0
from vedastr.utils import Registry

RECTIFICATORS = Registry('Rectificator')
示例#19
0
from vedastr.utils import Registry

METRICS = Registry('metrics')
示例#20
0
from vedastr.utils import Registry

RUNNERS = Registry('runner')
示例#21
0
from vedastr.utils import Registry

ENHANCE_MODULES = Registry('enhance_module')
示例#22
0
from vedastr.utils import Registry

COMPONENT = Registry('component')
BODIES = Registry('body')