rscd/models/decoderheads/none.py

import torch

class none_class(torch.nn.Module):
    def forward(self, x):
        return x