File size: 290 Bytes
74b17e0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import torch.nn as nn

from . import register_connector
from .base import Connector


    
@register_connector('identity')    
class IdentityConnector(Connector):
    def __init__(self, config=None):
        super().__init__()
        self._connector = nn.Identity()