Method forward

models/networks.py:15–16

(self, x)

from models.att import LocalDynamics
class Identity(nn.Module):
    def forward(self, x):
        return x

def get_norm_layer(norm_type='instance'):
    """Return a normalization layer