Source code for xmodaler.modeling.encoder.encoder

# Copyright 2021 JD.com, Inc., JD AI
"""
@author: Yehao Li
@contact: yehaoli.sysu@gmail.com
"""
import torch
from torch import nn

from xmodaler.config import configurable
from xmodaler.config import CfgNode as CN
from xmodaler.config import kfg
from .build import ENCODER_REGISTRY

__all__ = ["Encoder"]

[docs]@ENCODER_REGISTRY.register()
class Encoder(nn.Module):
[docs]    @configurable
    def __init__(self):
        super(Encoder, self).__init__()
        
[docs]    @classmethod
    def from_config(cls, cfg):
        return {}

[docs]    @classmethod
    def add_config(cls, cfg):
        pass

[docs]    def forward(self, batched_inputs, mode=None):
        return {}