Example #1
0
from vedastr.utils import Registry

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

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

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

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

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

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

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

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

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

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

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

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

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

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

UTILS = Registry('utils')
Example #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)
Example #17
0
from torch.optim import lr_scheduler

from vedastr.utils import Registry


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

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

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

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

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

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